Fix: sample option selection

app.py

@@ -28,15 +28,19 @@ sample_images = [
     os.path.join(os.getcwd(), "data/xai/sample2.jpg")
 ]
 
-# Interface definition
 interface = gr.Interface(
     fn=process_image,
     inputs=[
         gr.Radio(
-            label="Select Sample Image", 
-            choices=sample_images,  # List of sample images
-            value=sample_images[0],  # Default value to highlight the first image
-            type="pil",  # Type of the input value (image)
+            label="Select Sample Image or Upload Image", 
+            choices=["None"] + sample_images,  # Add 'None' option for uploading image
+            value="None",  # Default option is 'None'
+            type="value",  # We pass the file path or 'None' here
+        ),
+        gr.Image(
+            label="Upload Image", 
+            type="pil",  # Type 'pil' to work with PIL images
+            visible=False  # Initially hidden, will only be visible if 'None' is selected
         ),
         gr.CheckboxGroup(
             choices=["yolov5", "yolov8s"],
@@ -47,7 +51,7 @@ interface = gr.Interface(
     ],
     outputs=gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500),
     title="Explainable AI for YOLO Models",
-    description="Select a sample image and choose a model to visualize YOLO object detection with Grad-CAM.",
+    description="Select a sample image or upload your own image and choose a model to visualize YOLO object detection with Grad-CAM.",
     examples=sample_images,  # Example images to display
     live=True,
 )