Hugging Face's logo

Spaces:
suresh-subramanian
/
bean-classification
Runtime error

App Files Community
suresh-subramanian commited on
Commit
ac4b420
·
1 Parent(s): 2f2f139

Update app.py

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