Initial Commit

app.py

@@ -52,15 +52,16 @@ with my_app:
             with gr.Column():
                 with gr.Row():
                     img_source = gr.Image(label="Please select picture and click the part to segment", 
-                                          value='./images/truck.jpg', shape=(1024, 1024))
+                                          value='./images/truck.jpg').style(height=500, width=1000)
                 with gr.Row():
                     coords = gr.Label(label="Image Coordinate")
-                    model_choice = gr.Dropdown(['ViT-base', 'ViT-large', 'ViT-huge'], label='Model Backbone')
+                    with gr.Column():
+                        model_choice = gr.Dropdown(['ViT-base', 'ViT-large', 'ViT-huge'], label='Model Backbone')
+                        infer = gr.Button(label="Segment")
                 with gr.Row():
-                    infer = gr.Button(label="Segment")
                     score = gr.Label(label="Mask Confidence")
                 with gr.Row():
-                    img_output = gr.Image(label="Output Mask", shape=(1024, 1024))
+                    img_output = gr.Image(label="Output Mask").style(height=500, width=1000)
                     
         img_source.select(get_coords, [], coords)
         infer.click(