Update app.py

app.py

@@ -109,12 +109,12 @@ def process_image(image, model_choice):
 # Create the Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Document Segmentation Demo (ZeroGPU)")
-    
+
     with gr.Row():
-        model_choice = gr.Dropdown(["YOLOv8 Model", "DLA Model"], label="Select Model", value="YOLOv8 Model")
         input_image = gr.Image(type="pil", label="Upload Image")
-    
-    output_image = gr.Image(type="pil", label="Annotated Image")
+        output_image = gr.Image(type="pil", label="Annotated Image")
+
+    model_choice = gr.Dropdown(["YOLOv8 Model", "DLA Model"], label="Select Model", value="YOLOv8 Model", scale=0.5)
     output_text = gr.Textbox(label="Detected Areas and Labels")
     
     btn = gr.Button("Run Document Segmentation")