Update app.py

app.py

@@ -46,11 +46,12 @@ def run_theia(image):
         gt=True,
         device='cuda',
     )
-    
-    theia_decode_results = (255.0 * theia_decode_results[0]).astype(np.uint8)
-    gt_decode_results = (255.0 * gt_decode_results[0]).astype(np.uint8)
 
-    return [(theia_decode_results, "Theia Results"), (gt_decode_results, "Ground Truth")]
+    _, 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:, :]
+
+    return [theia_decode_results, gt_decode_results]
 
 with gr.Blocks() as demo:
     gr.HTML(load_description("gradio_title.md"))
@@ -62,8 +63,14 @@ with gr.Blocks() as demo:
             submit_button = gr.Button("Submit")
         
         with gr.Column():
-            output_gallery = gr.Gallery(label="Input, DINOv2, SAM, Depth Anything", type="numpy")
+            with gr.Row():
+                gr.Markdown("###Theia Results")
+                theia_output = gr.Image(label="DINOv2, SAM, Depth Anything", type="numpy")
+
+            with gr.Row():
+                gr.Markdown("###Ground Truth")
+                gt_output = gr.Image(label="DINOv2, SAM, Depth Anything", type="numpy")
 
-    submit_button.click(run_theia, inputs=input_image, outputs=output_gallery)
+    submit_button.click(run_theia, inputs=input_image, outputs=[theia_output, gt_output])
 
 demo.launch()