Update app.py

app.py

@@ -267,8 +267,12 @@ def get_mesh(image, depth):
 def loadurl(url):
     return url
 
-def select_index(glr):
-    return gr.Gallery(selected_index=glr.selected_index)
+def select_index(toggle, glr):
+    if toggle == True:
+        output_depth.selected_index = output_frame.selected_index
+    else:
+        output_frame.selected_index = output_depth.selected_index
+    return
 
 
 css = """
@@ -319,8 +323,8 @@ with gr.Blocks(css=css) as demo:
             input_url.change(fn=loadurl, inputs=[input_url], outputs=[input_video])
             output_frame = gr.Gallery(label="Frame", type='numpy', preview=True, columns=8192)
             output_depth = gr.Gallery(label="Depth", type='numpy', preview=True, columns=8192)
-            output_frame.select(fn=select_index, inputs=[output_frame], outputs=[output_depth])
-            output_depth.select(fn=select_index, inputs=[output_depth], outputs=[output_frame])
+            output_frame.select(fn=select_index, inputs=[True, output_frame], outputs=[])
+            output_depth.select(fn=select_index, inputs=[False, output_depth], outputs=[])
             submit = gr.Button("Submit")
         with gr.Column():
             model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl")], type="value", value="vits", label='Model Type')