Update app.py

app.py

@@ -188,11 +188,18 @@ def predict_beetle(img):
 
 
 # gradio app
-css = """
-button {
-    width: auto;  /* Set your desired width */
-}
-"""
+# css = """
+# button {
+#     width: auto;  /* Set your desired width */
+# }
+# """
+
+# 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'))]
+
 
 with gr.Blocks() as demo: # css=css in the brackets
     gr.Markdown("<h1><center>Bark Beetle Classification<h1><center>")
@@ -205,5 +212,9 @@ 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
+    
     btn.click(predict_beetle, inputs, gallery)
 demo.launch(debug=True, show_error=True)