Update app.py

app.py

@@ -11,6 +11,7 @@ from torchvision.transforms import Compose
 import trimesh
 from geometry import create_triangles
 import tempfile
+from functools import partial
 import spaces
 from zipfile import ZipFile
 
@@ -209,6 +210,7 @@ def pano_depth_to_world_points(depth):
 def get_mesh(image, depth, keep_edges=False):
     image.thumbnail((1024,1024))  # limit the size of the image
     pts3d = pano_depth_to_world_points(depth)
+    print('radius from depth - ok')
 
     # Create a trimesh mesh from the points
     # Each pixel is connected to its 4 neighbors
@@ -220,13 +222,16 @@ def get_mesh(image, depth, keep_edges=False):
         triangles = create_triangles(image.shape[0], image.shape[1])
     else:
         triangles = create_triangles(image.shape[0], image.shape[1], mask=~depth_edges_mask(depth))
+    print('triangles - ok')
     colors = image.reshape(-1, 3)
     mesh = trimesh.Trimesh(vertices=verts, faces=triangles, vertex_colors=colors)
+    print('mesh - ok')
 
     # Save as glb
     glb_file = tempfile.NamedTemporaryFile(suffix='.glb', delete=False)
     glb_path = glb_file.name
     mesh.export(glb_path)
+    print('file - ok')
     return glb_path
 
 def loadurl(url):
@@ -299,7 +304,7 @@ with gr.Blocks(css=css) as demo:
         return output_video_path
 
     submit.click(on_submit, inputs=[input_video, model_type], outputs=[processed_video, processed_zip, output_frame, output_depth])
-    render.click(get_mesh, inputs=[output_frame, output_depth, checkbox], outputs=[result])
+    render.click(partial(get_mesh), inputs=[output_frame, output_depth, checkbox], outputs=[result])
 
     example_files = os.listdir('examples')
     example_files.sort()