Spaces:

shatrunjai
/

leaf_classifier

Sleeping

shatrunjai commited on Apr 30, 2023

Commit

2e922d4

1 Parent(s): b3bb3d2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
 model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = dataset['train'].features['labels'].names
-image_processor = ViTImageProcessor.from_pretrained('google/vit-base-patch16-224')
 def classify(im):
   features = image_processor(im, return_tensors='pt')

 model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = dataset['train'].features['labels'].names
 def classify(im):
   features = image_processor(im, return_tensors='pt')