tadeyina commited on
Commit
11ba4f9
1 Parent(s): 3aac214

Update 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("s")
9
 
10
- model = AutoModelForImageClassification.from_pretrained("Blaise-MR/diseaseClassifier")
11
 
12
  labels = dataset['train'].features['labels'].names
13
 
5
 
6
  dataset = datasets.load_dataset('beans')
7
 
8
+ extractor = AutoFeatureExtractor.from_pretrained("tadeyina/BeanLeaf")
9
 
10
+ model = AutoModelForImageClassification.from_pretrained("tadeyina/BeanLeaf")
11
 
12
  labels = dataset['train'].features['labels'].names
13