Spaces:

brendenc
/

Plant_Disease_Detector

Build error

brendenc commited on Sep 26, 2022

Commit

bf0d6f7

•

1 Parent(s): c8189b0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,6 +8,7 @@ extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
 model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = dataset['train'].features['labels'].names
 def classify(im):
   features = feature_extractor(im, return_tensors='pt')

 model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = dataset['train'].features['labels'].names
+example_imgs = ["example_0.jpg", "example_1.jpg","example_2.jpg"]
 def classify(im):
   features = feature_extractor(im, return_tensors='pt')