Update app.py

app.py

@@ -197,11 +197,14 @@ def predict_beetle(img):
 
 # 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"],
+    [os.path.join(sample_images_dir, "example1.jpg")],
+    [os.path.join(sample_images_dir, "example2.jpg")],
+    [os.path.join(sample_images_dir, "example3.jpg")],
+    [os.path.join(sample_images_dir, "mixed.jpg")]
+    
 ]
+# Corresponding labels for the example images
+example_labels = ["Example Beetle 1", "Example Beetle 2", "Example Beetle 3", "Example Beetles Mixed"]
 
 with gr.Blocks() as demo: # css=css in the brackets
     gr.Markdown("<h1><center>Bark Beetle Classification<h1><center>")
@@ -215,8 +218,13 @@ with gr.Blocks() as demo: # css=css in the brackets
         # Set the gallery layout and height directly in the constructor
         gallery = gr.Gallery(label="Show images", show_label=True, elem_id="gallery", columns=8, height="auto")
 
-    # Add examples for easy access
-    demo.examples = example_images
+    # Add examples with labels
+    gr.Examples(
+        examples=example_images,
+        inputs=inputs,
+        examples_per_page=3,
+        example_labels=example_labels
+    )
     
     btn.click(predict_beetle, inputs, gallery)
 demo.launch(debug=True, show_error=True)