Spaces:

hari-whitebalance
/

corise-week-1-project

Sleeping

hari-whitebalance commited on Sep 26, 2022

Commit

195d2f3

•

1 Parent(s): 498b29d

reference was incorrect

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')
   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()