Fix: interface layout

app.py

@@ -139,9 +139,14 @@ with gr.Blocks() as interface:
         inputs=sample_selection,
         outputs=sample_display,
     )
-    
-    # Adding a placeholder for spacing
-    gr.Markdown(" ")
+
+    # Adding an upload placeholder
+    gr.Markdown("**Or upload your own image:**")
+    upload_image = gr.Image(
+        label="Upload an Image",
+        type="file",
+        tool="editor",  # Optional: allows basic editing like cropping
+    )
 
     selected_models = gr.CheckboxGroup(
         choices=["yolov5", "yolov8s"],
@@ -152,7 +157,7 @@ with gr.Blocks() as interface:
 
     gr.Button("Run").click(
         fn=process_image,
-        inputs=[sample_selection, selected_models],
+        inputs=[sample_selection, upload_image, selected_models],  # Include uploaded image
         outputs=result_gallery,
     )