Update app.py

app.py

@@ -40,10 +40,10 @@ def run_theia(image):
         device='cuda',
     )
     
-    vis = np.vstack([theia_decode_results[0], gt_decode_results[0]])
-    vis = (255.0 * vis).astype(np.uint8)
+    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 vis
+    return [(theia_decode_results, "Theia Results"), (gt_decode_results, "Ground Truth")]
 
-demo = gr.Interface(fn=run_theia, inputs=gr.Image(type="pil"), outputs=gr.Image(type="numpy"))
+demo = gr.Interface(fn=run_theia, inputs=gr.Image(type="pil"), outputs=gr.Gallery(label="Input, DINOv2, SAM, Depth Anything", type="numpy"))
 demo.launch()