Hugging Face's logo

Spaces:
Blaise-MR
/
BeanPlantDiseaseClassifier
Runtime error

App Files Community
Blaise-MR commited on
Commit
ff56218
•
1 Parent(s): d03f92c

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,9 +5,9 @@ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
5
 
6
  dataset = datasets.load_dataset('beans')
7
 
8
- extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
9
 
10
- model = AutoModelForImageClassification.from_pretrained("saved_model_files")
11
 
12
  labels = dataset['train'].features['labels'].names
13
 
5
 
6
  dataset = datasets.load_dataset('beans')
7
 
8
+ extractor = AutoFeatureExtractor.from_pretrained("./")
9
 
10
+ model = AutoModelForImageClassification.from_pretrained("./")
11
 
12
  labels = dataset['train'].features['labels'].names
13