Update app.py

app.py

@@ -68,13 +68,21 @@ def predict(image):
     except Exception as e:
         return {'error': str(e)}
 
-# Set up Gradio interface
+# Example input images (provide paths or URLs)
+example_images = [
+    "https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/viewer?row=0&image-viewer=52B421CB70A43313B278D5DD2C58CECE56343012",  # Replace with the actual paths to your example images
+    "https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/viewer/default/train?p=2&row=201&image-viewer=558EA847F2267CECF4E2CFF6352F9D8888E9A72F",
+    "https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/viewer/default/train?p=2&row=201&image-viewer=8FBAF2C52C256A392660811C5659788734821C3A"
+]
+
+# Set up Gradio interface with examples
 iface = gr.Interface(
     fn=predict,
     inputs=gr.inputs.Image(type="pil", label="Upload an image"),
     outputs=gr.outputs.Label(num_top_classes=3, label="Predicted class and confidence"),
     title="Bacterial Morphology Classifier",
-    description="Upload an image of a bacterial sample to classify it as 'cocci', 'bacilli', or 'spirilla'."
+    description="Upload an image of a bacterial sample to classify it as 'cocci', 'bacilli', or 'spirilla'.",
+    examples=example_images  # Provide the example image paths
 )
 
 # Launch the app