Spaces:

theaiinstitute
/

theia

Running on Zero

bmay commited on Aug 1, 2024

Commit

044c290

verified ·

1 Parent(s): 6273591

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,19 +66,13 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### Input Image")
-            input_image = gr.Image(type="pil", label=None)
             submit_button = gr.Button("Submit")
         with gr.Column():
-            gr.Markdown("### DINOv2")
-            dinov2_output = gr.Gallery(label=None, type="numpy")
-            gr.Markdown("### SAM")
-            sam_output = gr.Gallery(label=None, type="numpy")
-            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])

     with gr.Row():
         with gr.Column():
+            input_image = gr.Image(label="Input Image", type="pil")
             submit_button = gr.Button("Submit")
         with gr.Column():
+            dinov2_output = gr.Gallery(label="DINOv2", type="numpy")
+            sam_output = gr.Gallery(label="SAM", type="numpy")
+            depth_anything_output = gr.Gallery(label="Depth-Anything", type="numpy")
     submit_button.click(run_theia, inputs=input_image, outputs=[dinov2_output, sam_output, depth_anything_output])