Spaces:

RKoops
/

BeanLeafApp

Runtime error

RKoops commited on Sep 27, 2022

Commit

3140f1a

1 Parent(s): ee89c9b

Update used model name

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,8 @@ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 dataset = datasets.load_dataset("beans")
 labels = dataset["train"].features["labels"].names
-extractor = AutoFeatureExtractor.from_pretrained("RKoops/BeanLeaf")
-model = AutoModelForImageClassification.from_pretrained("RKoops/BeanLeaf")
 def classify(im):

 dataset = datasets.load_dataset("beans")
 labels = dataset["train"].features["labels"].names
+extractor = AutoFeatureExtractor.from_pretrained("RKoops/BeanLeafClassifier")
+model = AutoModelForImageClassification.from_pretrained("RKoops/BeanLeafClassifier")
 def classify(im):