Update app.py

app.py

@@ -31,11 +31,11 @@ def process_image(image):
 # Create the Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Document Segmentation Demo (ZeroGPU)")
-    # Input Components
-    input_image = gr.Image(type="pil", label="Upload Image")
 
-    # Output Components
-    output_image = gr.Image(type="pil", label="Annotated Image")
+    with gr.Row():
+        input_image = gr.Image(type="pil", label="Upload Image")
+        output_image = gr.Image(type="pil", label="Annotated Image")
+
     output_text = gr.Textbox(label="Detected Areas and Labels")
 
     # Button to trigger inference