Fix: gradio interface

app.py

@@ -28,17 +28,13 @@ sample_images = {
 }
 
 
-def load_sample(sample_name):
-    if sample_name and sample_name in sample_images:
-        return Image.open(sample_images[sample_name])
-    return None
-
+"""
 interface = gr.Interface(
     fn=process_image,
     inputs=[
         gr.Image(type="pil", label="Upload an Image"),
         gr.CheckboxGroup(
-            choices=["yolov3", "yolov8s"],
+            choices=["yolov5", "yolov8s"],
             value=["yolov5"],  # Set the default value (YOLOv5 checked by default)
             label="Select Model(s)",
         ),
@@ -65,3 +61,39 @@ def main_logic(uploaded_image, selected_models, sample_selection):
     return process_image(image, selected_models)
 
 interface.launch()
+"""
+
+def load_sample_image(sample_name):
+    if sample_name in sample_images:
+        return Image.open(sample_images[sample_name])
+    return None
+
+with gr.Blocks() as interface:
+    gr.Markdown("# Visualizing Key Features with Explainable AI")
+    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")
+        sample_selection = gr.Dropdown(
+            choices=list(sample_images.keys()),
+            label="Select a Sample Image",
+            type="value",
+        )
+        sample_display = gr.Image(label="Sample Image Preview")
+        sample_selection.change(fn=load_sample_image, inputs=sample_selection, outputs=sample_display)
+
+    selected_models = gr.CheckboxGroup(
+        choices=["yolov3", "yolov8s"],
+        value=["yolov5"],  # Default model
+        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,
+    )
+
+interface.launch()