Update app.py

app.py

@@ -225,7 +225,8 @@ def pano_depth_to_world_points(depth, scale):
 def rgb2gray(rgb):
     return np.dot(rgb[...,:3], [0.333, 0.333, 0.333])
 
-def get_mesh(image, depth, fnum):
+def get_mesh(image, depth):
+    fnum = output_frame.selected_index
     gdepth = rgb2gray(depth[fnum][0])
     print('depth to gray - ok')
     points = pano_depth_to_world_points(gdepth, 1)
@@ -338,7 +339,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(partial(get_mesh), inputs=[output_frame, output_depth, frame_n], outputs=[result])
+    render.click(partial(get_mesh), inputs=[output_frame, output_depth], outputs=[result])
 
     example_files = os.listdir('examples')
     example_files.sort()