Hugging Face's logo

Spaces:
RKoops
/
BeanLeafApp
Runtime error

App Files Community
RKoops commited on
Commit
3140f1a
•
1 Parent(s): ee89c9b

Update used model name

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