Hugging Face's logo

Spaces:
KhadijaAsehnoune12
/
Orange-Disease-Classifier
Sleeping

App Files Community
KhadijaAsehnoune12 commited on
Commit
efac948
·
verified ·
1 Parent(s): f4f732b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import ViTFeatureExtractor, ViTForImageClassification
4
  from PIL import Image
5
 
6
  # Définir le modèle et le feature extractor
7
- model_name ="KhadijaAsehnoune12/OrangeLeafDisease-Detector"
8
  model = ViTForImageClassification.from_pretrained(model_name, num_labels=10, ignore_mismatched_sizes=True)
9
  feature_extractor = ViTFeatureExtractor.from_pretrained(model_name)
10
 
 
4
  from PIL import Image
5
 
6
  # Définir le modèle et le feature extractor
7
+ model_name ="KhadijaAsehnoune12/ViTOrangeLeafDiseaseClassifier"
8
  model = ViTForImageClassification.from_pretrained(model_name, num_labels=10, ignore_mismatched_sizes=True)
9
  feature_extractor = ViTFeatureExtractor.from_pretrained(model_name)
10