Fix: model dependency + default sample

app.py

@@ -21,6 +21,7 @@ def process_image(image, yolo_versions=["yolov5"]):
             result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
     return result_images
 
+
 sample_images = {
     "Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
     "Sample 2":  os.path.join(os.getcwd(), "data/xai/sample2.jpg"),
@@ -31,17 +32,20 @@ def load_sample_image(sample_name):
         return Image.open(image_path)
     return None
 
+# Default sample image
 default_sample_image = load_sample_image("Sample 1")
+
+# Gradio interface
 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.")
 
     with gr.Row():
-        uploaded_image = gr.Image(type="pil", label="Upload an Image", value = None)
+        uploaded_image = gr.Image(type="pil", label="Upload an Image")
         sample_selection = gr.Dropdown(
             choices=list(sample_images.keys()),
             label="Select a Sample Image",
-            type="Sample 1",
+            type="value",
         )
         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)
@@ -53,11 +57,10 @@ with gr.Blocks() as interface:
     )
 
     result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
-    print(uploaded_image)
-    print(sample_display)
+
     gr.Button("Run").click(
         fn=process_image,
-        inputs=[uploaded_image, selected_models],        
+        inputs=[uploaded_image, selected_models, sample_display],
         outputs=result_gallery,
     )