Update app.py

app.py

@@ -59,19 +59,16 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
+            gr.Markdown("### Input Image")
             input_image = gr.Image(type="pil", label="Input Image")
             submit_button = gr.Button("Submit")
         
         with gr.Column():
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown("### Theia Results")
-                    theia_output = gr.Image(label="DINOv2, SAM, Depth Anything", type="numpy")
+            gr.Markdown("### Theia Results")
+            theia_output = gr.Image(label="DINOv2, SAM, Depth Anything", type="numpy")
 
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown("### Ground Truth")
-                    gt_output = gr.Image(label="DINOv2, SAM, Depth Anything", type="numpy")
+            gr.Markdown("### Ground Truth")
+            gt_output = gr.Image(label="DINOv2, SAM, Depth Anything", type="numpy")
 
     submit_button.click(run_theia, inputs=input_image, outputs=[theia_output, gt_output])