Update app.py

app.py

@@ -74,10 +74,10 @@ def app():
         with gr.Row():
             with gr.Column():
                 with gr.Row():
-                    raw_image = gr.Image(type="pil", label="Image", visible=True, interactive=True)
+                    raw_image = gr.Image(type="pil", label="Image", visible=True, interactive=True, height=600)
                     box_image = ImagePrompter(type="pil", label="DrawBox", visible=False, interactive=True)
                     mask_image = gr.ImageEditor(type="pil", label="DrawMask", visible=False, interactive=True, layers=False, canvas_size=(640, 640))
-                    target_image = gr.Image(type="pil", label="Target Image", visible=False, interactive=True)
+                    target_image = gr.Image(type="pil", label="Target Image", visible=False, interactive=True, height=600)
                 
                 yoloe_infer = gr.Button(value="Detect & Segment Objects")
                 prompt_type = gr.Textbox(value="Text", visible=False)
@@ -133,7 +133,7 @@ def app():
                 )
 
             with gr.Column():
-                output_image = gr.Image(type="numpy", label="Annotated Image", visible=True)
+                output_image = gr.Image(type="numpy", label="Annotated Image", visible=True, height=600)
         
         def update_text_image_visibility():
             return gr.update(value="Text"), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)