Spaces:

deman539
/

beans-dataset-classification

Runtime error

Dhritiman Sagar commited on Sep 22, 2022

Commit

fc5989d

•

1 Parent(s): 55bead6

Fix bug in app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = dataset['train'].features['labels'].names
 def classify(im):
-  features = feature_extractor(im, return_tensors='pt')
   logits = model(features["pixel_values"])[-1]
   probability = torch.nn.functional.softmax(logits, dim=-1)
   probs = probability[0].detach().numpy()

 labels = dataset['train'].features['labels'].names
 def classify(im):
+  features = extractor(im, return_tensors='pt')
   logits = model(features["pixel_values"])[-1]
   probability = torch.nn.functional.softmax(logits, dim=-1)
   probs = probability[0].detach().numpy()