Fix: argument mismatch

app.py

@@ -26,19 +26,25 @@ def process_image(image, yolo_versions=["yolov5"]):
 interface = gr.Interface(
     fn=process_image,
     inputs=[
-        gr.Image(type="pil", label="Upload an Image"),
+        gr.Image(
+            type="pil", 
+            label="Upload an Image", 
+            interactive=True,
+            value=None  # This is where the sample image will be loaded when selected
+        ),
         gr.CheckboxGroup(
             choices=["yolov5", "yolov8s"],
-            value=["yolov5"],  # Set default selection to YOLOv5
+            value=["yolov5"],  # Ensure default value is set correctly
             label="Select Model(s)",
+            type="value",  # Ensure the value is passed as a list of selected models
         ),
     ],
     outputs=gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500),
     title="Explainable AI for YOLO Models",
     description="Upload an image or select a sample to visualize YOLO object detection with Grad-CAM.",
     examples=[
-        [os.path.join(os.getcwd(), "data/xai/sample1.jpeg")],  
-        [os.path.join(os.getcwd(), "data/xai/sample2.jpg")],  
+        [os.path.join(os.getcwd(), "data/xai/sample1.jpeg"), "yolov5"],  
+        [os.path.join(os.getcwd(), "data/xai/sample2.jpg"), "yolov8s"],  
     ],
     live=True  
 )