Update app.py

app.py

@@ -5,11 +5,12 @@ from PIL import Image
 # Initialize the plant disease classification pipeline
 # You can replace the model with any fine-tuned plant disease model hosted on Hugging Face
 plant_disease_classifier = pipeline(
-    task="image-classification",  
-    model="nielsr/plant-disease-model",  
-    top_k=3  # return top 3 predictions
+    task="image-classification",
+    model="wambugu71/crop_leaf_diseases_vit",
+    top_k=3
 )
 
+
 def diagnose_plant_health(image: Image.Image):
     """
     Takes a PIL Image of a plant leaf and returns: