Update app.py

app.py

@@ -194,12 +194,13 @@ def predict_beetle(img):
 # }
 # """
 
-# Sample images directory
-sample_images_dir = "example_images"
-
-# Collect sample image paths (these should be paths to any images you want to use as examples)
-example_images = [[os.path.join(sample_images_dir, img)] for img in os.listdir(sample_images_dir) if img.endswith(('.png', '.jpg', '.jpeg'))]
-
+# Sample images with labels
+example_images = [
+    [os.path.join("example_images", "example1.jpg"), "Example Beetles 1"],
+    [os.path.join("example_images", "example2.jpg"), "Example Beetles 2"],
+    [os.path.join("example_images", "example3.jpg"), "Example Beetles 3"],
+    [os.path.join("example_images", "mixed.jpg"), "Example Beetles Mixed"],
+]
 
 with gr.Blocks() as demo: # css=css in the brackets
     gr.Markdown("<h1><center>Bark Beetle Classification<h1><center>")