Update app.py

app.py

@@ -33,7 +33,7 @@ class BacterialMorphologyClassifier(nn.Module):
         return x
 
 # Load the model
-MODEL_PATH = "https://huggingface.co/yolac/BacterialMorphologyClassification/resolve/main/model.pth"
+MODEL_PATH = "model.pth"
 model = BacterialMorphologyClassifier()
 
 try:
@@ -78,6 +78,13 @@ def predict(image):
     except Exception as e:
         return f"Error: {str(e)}"
 
+# Define example images
+examples = [
+    ["https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20290.jpg"],
+    ["https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20565.jpg"],
+    ["https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%208.jpg"],
+]
+
 # Set up Gradio interface
 interface = gr.Interface(
     fn=predict,
@@ -85,6 +92,7 @@ interface = gr.Interface(
     outputs=gr.Text(label="Prediction"),
     title="Bacterial Morphology Classification",
     description="Upload an image of bacteria to classify it as cocci, bacilli, or spirilla.",
+    examples=examples,
 )
 
 # Launch the app