Update app.py

app.py

@@ -48,10 +48,16 @@ def run_theia(image):
     )
 
     _, width, _ = theia_decode_results[0].shape
-    theia_decode_results = (255.0 * theia_decode_results[0]).astype(np.uint8)[:, width // 4:, :]
-    gt_decode_results = (255.0 * gt_decode_results[0]).astype(np.uint8)[:, width // 4:, :]
+    theia_decode_dino = (255.0 * theia_decode_results[0]).astype(np.uint8)[:, width // 4 : 2 * width // 4, :]
+    theia_decode_sam = (255.0 * theia_decode_results[0]).astype(np.uint8)[:, 2 * width // 4 : 3 * width // 4, :]
+    theia_decode_depth = (255.0 * theia_decode_results[0]).astype(np.uint8)[:, 3 * width // 4 :, :]
+    gt_dino = (255.0 * theia_decode_results[0]).astype(np.uint8)[:, width // 4 : 2 * width // 4, :]
+    gt_sam = (255.0 * theia_decode_results[0]).astype(np.uint8)[:, 2 * width // 4 : 3 * width // 4, :]
+    gt_depth = (255.0 * theia_decode_results[0]).astype(np.uint8)[:, 3 * width // 4 :, :]
 
-    return [theia_decode_results, gt_decode_results]
+    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
 
 with gr.Blocks() as demo:
     gr.HTML(load_description("gradio_title.md"))
@@ -60,15 +66,15 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             gr.Markdown("### Input Image")
-            input_image = gr.Image(type="pil", label="Input Image")
+            input_image = gr.Image(type="pil", label=None)
             submit_button = gr.Button("Submit")
         
         with gr.Column():
             gr.Markdown("### Theia Results")
-            theia_output = gr.Image(label="DINOv2, SAM, Depth Anything", type="numpy")
+            theia_output = gr.Gallery(label=None, type="numpy")
 
             gr.Markdown("### Ground Truth")
-            gt_output = gr.Image(label="DINOv2, SAM, Depth Anything", type="numpy")
+            gt_output = gr.Gallery(label=None, type="numpy")
 
     submit_button.click(run_theia, inputs=input_image, outputs=[theia_output, gt_output])