Spaces:

cupkake14
/

bean_vit_classifier

Build error

cupkake14 commited on Sep 22, 2022

Commit

d2cb445

1 Parent(s): 7aa6962

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = dataset['train'].features['labels'].names
 def classify(im):
-  features = feature_extractor(im, return_tensors='pt')
   with torch.no_grad():
     logits = model(**features).logits
   probability = torch.nn.functional.softmax(logits, dim=-1)

 labels = dataset['train'].features['labels'].names
 def classify(im):
+  features = extractor(im, return_tensors='pt')
   with torch.no_grad():
     logits = model(**features).logits
   probability = torch.nn.functional.softmax(logits, dim=-1)