Spaces:

srishti-hf1110
/

Bean-plan-classifier

Sleeping

srishti-hf1110 commited on Sep 26, 2022

Commit

50ba0a4

1 Parent(s): c67444b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,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["pixel_values"])[-1]
   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["pixel_values"])[-1]
   probability = torch.nn.functional.softmax(logits, dim=-1)