Fix: argument mismatch

app.py

@@ -22,30 +22,34 @@ def process_image(image, yolo_versions=["yolov5"]):
             result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
     return result_images
 
+# Paths to sample images
+sample_images = [
+    os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
+    os.path.join(os.getcwd(), "data/xai/sample2.jpg")
+]
 
+# Interface definition
 interface = gr.Interface(
     fn=process_image,
     inputs=[
-        gr.Image(
-            type="pil", 
-            label="Upload an Image", 
-            interactive=True,
-            value=None,  # Default value set to None
+        gr.Gallery(
+            label="Select Sample Image", 
+            value=sample_images[0],  # Default value to highlight the first image
+            choices=sample_images,  # List of sample images
+            type="pil",
+            interactive=True  # Allow user to click and select an image
         ),
         gr.CheckboxGroup(
             choices=["yolov5", "yolov8s"],
-            value=["yolov5"],  # Default value set
+            value=["yolov5"],  # Default model selected
             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")],  
-    ],
+    description="Select a sample image and choose a model to visualize YOLO object detection with Grad-CAM.",
+    examples=sample_images,  # Example images to display
     live=True,
 )