Fix: model dependency + default sample

app.py

@@ -26,18 +26,13 @@ sample_images = {
     "Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
     "Sample 2":  os.path.join(os.getcwd(), "data/xai/sample2.jpg"),
 }
-
 def load_sample_image(sample_name):
-    if sample_name in sample_images:
-        try:
-            return Image.open(sample_images[sample_name])  # Load and return the image
-        except Exception as e:
-            print(f"Error loading image: {e}")
-            return None
+    image_path = sample_images.get(sample_name)
+    if image_path and os.path.exists(image_path):
+        return Image.open(image_path)
     return None
-   
-default_sample_image = load_sample_image("Sample 1")
 
+default_sample_image = load_sample_image("Sample 1")
 with gr.Blocks() as interface:
     gr.Markdown("# XAI: Upload an image to visualize object detection of your models..")
     gr.Markdown("Upload an image or select a sample image to visualize object detection.")
@@ -49,7 +44,7 @@ with gr.Blocks() as interface:
             label="Select a Sample Image",
             type="value",
         )
-        sample_display = gr.Image(label="Sample Image Preview",  value=default_sample_image)
+        sample_display = gr.Image(label="Sample Image Preview", value=default_sample_image)
         sample_selection.change(fn=load_sample_image, inputs=sample_selection, outputs=sample_display)
 
     selected_models = gr.CheckboxGroup(
@@ -62,7 +57,7 @@ with gr.Blocks() as interface:
 
     gr.Button("Run").click(
         fn=process_image,
-        inputs=[uploaded_image, selected_models],
+        inputs=[uploaded_image, selected_models, sample_display],
         outputs=result_gallery,
     )