Fix: incorrect path

app.py

@@ -23,45 +23,9 @@ def process_image(image, yolo_versions=["yolov5"]):
 
 
 sample_images = {
-    "Sample 1": os.path.join(os.getcwd(), "/data/xai/sample1.jpeg"),
-    "Sample 2":  os.path.join(os.getcwd(), "/data/xai/sample2.jpg")
+    "Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
+    "Sample 2":  os.path.join(os.getcwd(), "data/xai/sample2.jpg"),
 }
-
-
-"""
-interface = gr.Interface(
-    fn=process_image,
-    inputs=[
-        gr.Image(type="pil", label="Upload an Image"),
-        gr.CheckboxGroup(
-            choices=["yolov5", "yolov8s"],
-            value=["yolov5"],  # Set the default value (YOLOv5 checked by default)
-            label="Select Model(s)",
-        ),
-        gr.Dropdown(
-            choices=list(sample_images.keys()),
-            label="Select a Sample Image",
-            type="value",
-            interactive=True,
-        ),
-    ],
-    outputs=gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500),
-    title="Visualising the key image features that drive decisions with our explainable AI tool.",
-    description="XAI: Upload an image or select a sample to visualize object detection of your models.",
-)
-
-def main_logic(uploaded_image, selected_models, sample_selection):
-    # If the user selects a sample image, use that instead of the uploaded one
-    if sample_selection:
-        image = load_sample(sample_selection)
-    else:
-        image = uploaded_image
-
-    # Call the processing function
-    return process_image(image, selected_models)
-
-interface.launch()
-"""
 def load_sample_image(sample_name):
     if sample_name in sample_images:
         try:
@@ -71,9 +35,6 @@ def load_sample_image(sample_name):
             return None
     return None
 
-
-
-# Gradio interface
 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.")