Update app_recon.py

app_recon.py

@@ -172,9 +172,9 @@ def reconstruction(image, files):
 
     normal_np[:, :, 0] *= -1
     _, surface, _,  _, _ = bilateral_normal_integration_function(normal_np, mask_output_temp, k=2, K=None, max_iter=100, tol=1e-4, cg_max_iter=5000, cg_tol=1e-3)
-    ply_path = os.path.join(dir_name, f"{name_base}_mask.ply")
+    ply_path = os.path.join(dir_name, f"{name_base}_recon.ply")
     surface.save(ply_path, binary=False)
-    return ply_path
+    return [ply_path]
 
 def run_demo():
 
@@ -271,8 +271,13 @@ def run_demo():
             recon_btn = gr.Button('Is there a salient foreground object? If yes, Click here to Reconstruct its 3D model.', variant='primary', interactive=True)
 
         with gr.Row():
-            reconstructed_3d = gr.Model3D(
-                label = 'Bini post-processed 3D model', height=320, interactive=False,
+            # reconstructed_3d = gr.Model3D(
+            #     label = 'Bini post-processed 3D model', height=320, interactive=False,
+            # )
+            reconstructed_3d = gr.Files(
+                label = "Bini post-processed 3D model (plyfile)",
+                elem_id = "download",
+                interactive=False,
             )