Fix: upload support

app.py

@@ -67,8 +67,12 @@ def load_sample_image(choice):
         raise ValueError("Invalid sample selection.")
 
 
-def process_image(choice, yolo_versions=["yolov5"]):
-    image = load_sample_image(choice)
+def process_image(sample_choice, uploaded_image, yolo_versions=["yolov5"]):
+    if uploaded_image is not None:
+        image = uploaded_image  # Use the uploaded image
+    else:
+        # Otherwise, use the selected sample image
+        image = load_sample_image(sample_choice)
     image = np.array(image)
     image = cv2.resize(image, (640, 640))
     result_images = []
@@ -81,6 +85,7 @@ def process_image(choice, yolo_versions=["yolov5"]):
             result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
     return result_images
 
+
 """
 import gradio as gr
 with gr.Blocks() as interface:
@@ -144,7 +149,7 @@ with gr.Blocks() as interface:
     gr.Markdown("**Or upload your own image:**")
     upload_image = gr.Image(
         label="Upload an Image",
-        type="filepath",  # Corrected 'type' parameter
+        type="filepath",  # Corrected type for file path compatibility
     )
 
     selected_models = gr.CheckboxGroup(
@@ -156,8 +161,8 @@ with gr.Blocks() as interface:
 
     gr.Button("Run").click(
         fn=process_image,
-        inputs=[upload_image, selected_models],  # Include uploaded image
+        inputs=[sample_selection, upload_image, selected_models],  # Include both options
         outputs=result_gallery,
     )
 
-interface.launch()
+interface.launch(share=True)