leonelhs commited on
Commit
f86c058
1 Parent(s): 88e3c7c

app changes

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -37,8 +37,7 @@ def make_bundle(image, name):
37
  def predict(img: np.ndarray):
38
  bundle = make_bundle(img, "Model3D")
39
  dataset = EvalWMetaDataset(opts, [bundle])
40
- img, model = reconstructor.evaluate(dataset)
41
- return img, model, model
42
 
43
 
44
  footer = r"""
@@ -59,10 +58,9 @@ with gr.Blocks(title="PIFuHD") as app:
59
  with gr.Column():
60
  output_obj = gr.Model3D(label="Output model")
61
  output_img = gr.Image(type="filepath", label="Output image")
62
- output_file = gr.File(label="Download 3D Model")
63
- gr.ClearButton(components=[input_img, output_img, output_obj, output_file], variant="stop")
64
 
65
- run_btn.click(predict, [input_img], [output_img, output_obj, output_file])
66
 
67
  with gr.Row():
68
  blobs = [[f"examples/{x:02d}.png"] for x in range(1, 4)]
 
37
  def predict(img: np.ndarray):
38
  bundle = make_bundle(img, "Model3D")
39
  dataset = EvalWMetaDataset(opts, [bundle])
40
+ return reconstructor.evaluate(dataset)
 
41
 
42
 
43
  footer = r"""
 
58
  with gr.Column():
59
  output_obj = gr.Model3D(label="Output model")
60
  output_img = gr.Image(type="filepath", label="Output image")
61
+ gr.ClearButton(components=[input_img, output_img, output_obj], variant="stop")
 
62
 
63
+ run_btn.click(predict, [input_img], [output_img, output_obj])
64
 
65
  with gr.Row():
66
  blobs = [[f"examples/{x:02d}.png"] for x in range(1, 4)]