Update app.py

app.py

@@ -55,9 +55,10 @@ def run_theia(image):
     gt_sam = (255.0 * gt_results[0]).astype(np.uint8)[:, 2 * width // 4 : 3 * width // 4, :]
     gt_depth = (255.0 * gt_results[0]).astype(np.uint8)[:, 3 * width // 4 :, :]
 
-    theia_output = [(theia_decode_dino, "DINOv2"), (theia_decode_sam, "SAM"), (theia_decode_depth, "Depth-Anything")]
-    gt_output = [(gt_dino, "DINOv2"), (gt_sam, "SAM"), (gt_depth, "Depth-Anything")]
-    return theia_output, gt_output
+    dinov2_output = [(theia_decode_dino, "Theia"), (gt_dino, "Ground Truth")]
+    sam_output = [(theia_decode_sam, "Theia"), (gt_sam, "Ground Truth")]
+    depth_anything_output = [(theia_decode_depth, "Theia"), (gt_depth, "Ground Truth")]
+    return dinov2_output, sam_output, depth_anything_output
 
 with gr.Blocks() as demo:
     gr.HTML(load_description("gradio_title.md"))
@@ -70,12 +71,15 @@ with gr.Blocks() as demo:
             submit_button = gr.Button("Submit")
         
         with gr.Column():
-            gr.Markdown("### Theia Results")
-            theia_output = gr.Gallery(label=None, type="numpy")
+            gr.Markdown("### DINOv2")
+            dinov2_output = gr.Gallery(label=None, type="numpy")
 
-            gr.Markdown("### Ground Truth")
-            gt_output = gr.Gallery(label=None, type="numpy")
+            gr.Markdown("### SAM")
+            sam_output = gr.Gallery(label=None, type="numpy")
 
-    submit_button.click(run_theia, inputs=input_image, outputs=[theia_output, gt_output])
+            gr.Markdown("### Depth-Anything")
+            depth_anything_output = gr.Gallery(label=None, type="numpy")
+
+    submit_button.click(run_theia, inputs=input_image, outputs=[dinov2_output, sam_output, depth_anything_output])
 
 demo.launch()