Fix: gradio interface

app.py

@@ -21,11 +21,12 @@ def process_image(image, yolo_versions=["yolov5"]):
             result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
     return result_images
 
-# Paths to sample images
-sample_images = [
-    os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
-    os.path.join(os.getcwd(), "data/xai/sample2.jpg")
-]
+
+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(sample_name):
     if sample_name and sample_name in sample_images: