Fix: interface update

app.py

@@ -10,7 +10,6 @@ from yolov8 import xai_yolov8s
 def process_image(image, yolo_versions=["yolov5"]):
     image = np.array(image)
     image = cv2.resize(image, (640, 640))
-
     result_images = []
     for yolo_version in yolo_versions:
         if yolo_version == "yolov5":
@@ -32,10 +31,8 @@ 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.")
@@ -52,16 +49,40 @@ with gr.Blocks() as interface:
 
     selected_models = gr.CheckboxGroup(
         choices=["yolov5", "yolov8s"],
-        value=["yolov5"],  # Default model
+        value=["yolov5"], 
         label="Select Model(s)",
     )
-
     result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
-
     gr.Button("Run").click(
         fn=process_image,
         inputs=[uploaded_image, selected_models],
         outputs=result_gallery,
     )
+"""
+
+with gr.Blocks() as interface:
+    gr.Markdown("# XAI: Upload an image to visualize object detection of your models..")
+    gr.Markdown("Select a sample image to visualize object detection.")
+
+    with gr.Row():
+        sample_selection = gr.RadioButtons(
+            choices=list(sample_images.keys()),
+            label="Select a Sample Image",
+            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)
+
+    selected_models = gr.CheckboxGroup(
+        choices=["yolov5", "yolov8s"],
+        value=["yolov5"], 
+        label="Select Model(s)",
+    )
+    result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
+    gr.Button("Run").click(
+        fn=process_image,
+        inputs=[sample_selection, selected_models],
+        outputs=result_gallery,
+    )
 
 interface.launch()