Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import ViTFeatureExtractor, ViTForImageClassification
 from PIL import Image
 
 # Define the model and feature extractor
-model_name = "KhadijaAsehnoune12/OrangeLeafDiseaseDetector"
+model_name = "KhadijaAsehnoune12/LeafDiseaseDetectorForOranges"
 model = ViTForImageClassification.from_pretrained(model_name, num_labels=10, ignore_mismatched_sizes=True)
 feature_extractor = ViTFeatureExtractor.from_pretrained(model_name)
 
@@ -51,4 +51,4 @@ gr.Interface(fn=predict,
              outputs="text",
              title="Citrus Disease Classification",
              description="Upload an image of a citrus leaf to classify its disease. The model is trained on the following diseases: Aleurocanthus spiniferus, Chancre citrique, Cochenille blanche, Dépérissement des agrumes, Feuille saine, Jaunissement des feuilles, Maladie de l'oïdium, Maladie du dragon jaune, Mineuse des agrumes, Trou de balle.",
-             examples=["maladie_du_dragon_jaune.jpg", "mineuse_des_agrumes.jpg"]).launch(share=True)
+             examples=["maladie_du_dragon_jaune.jpg", "mineuse_des_agrumes.jpg","feuille_saine.jpg"]).launch(share=True)