test12

app.py

@@ -134,9 +134,9 @@ def infer(in_pil_img, in_threshold=0.8, out_pil_img=None):
         # convert to PIL image
         out_pil_img =  Image.fromarray((input_img_overlay*255).astype(np.uint8))
 
-        return out_pil_img
+        return out_pil_img, all_mesh_paths
     else:
-        return None
+        return None, []
 
 
 with gr.Blocks(title="4DHumans", css=".gradio-container") as demo:
@@ -148,15 +148,14 @@ with gr.Blocks(title="4DHumans", css=".gradio-container") as demo:
             input_image = gr.Image(label="Input image", type="pil")
         with gr.Column():
             output_image = gr.Image(label="Reconstructions", type="pil")
-            # output_meshes = gr.File(label="3D meshes")
+            output_meshes = gr.File(label="3D meshes")
 
     gr.HTML("""<br/>""")
 
     with gr.Row():
         threshold = gr.Slider(0, 1.0, value=0.6, label='Detection Threshold')
         send_btn = gr.Button("Infer")
-        send_btn.click(fn=infer, inputs=[input_image, threshold], outputs=[output_image])
-        
+        send_btn.click(fn=infer, inputs=[input_image, threshold], outputs=[output_image, output_meshes])
 
     # gr.Examples([
     #     ['assets/test1.png', 0.6], 
@@ -167,20 +166,14 @@ with gr.Blocks(title="4DHumans", css=".gradio-container") as demo:
     #     ], 
     #     inputs=[input_image, threshold])
 
-    # with gr.Row():
-    #     gr.Examples([
-    #         ['assets/test1.png'], 
-    #         ['assets/test2.jpg'], 
-    #         ['assets/test3.jpg'], 
-    #         ['assets/test4.jpg'], 
-    #         ['assets/test5.jpg'], 
-    #         ], 
-
-    #         inputs=[input_image, 0.6],
-    #         outputs = [ 
-    #                 output_image
-    #             ],
-    #         )
+    gr.Examples([
+        ['assets/test1.png'], 
+        ['assets/test2.jpg'], 
+        ['assets/test3.jpg'], 
+        ['assets/test4.jpg'], 
+        ['assets/test5.jpg'], 
+        ], 
+        inputs=[input_image, 0.6])
 
 
 #demo.queue()