deman539 commited on
Commit
62cccf5
1 Parent(s): 5fec103

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ model = AutoModelForImageClassification.from_pretrained("saved_model_files")
8
  labels = dataset['train'].features['labels'].names
9
 
10
  def classify(input_image):
11
- features = image_processor(input_image, return_tensors='pt')
12
  logits = model(features["pixel_values"])[-1]
13
  probability = torch.nn.functional.softmax(logits, dim=-1)
14
  probs = probability[0].detach().numpy()
 
8
  labels = dataset['train'].features['labels'].names
9
 
10
  def classify(input_image):
11
+ features = extractor(input_image, return_tensors='pt')
12
  logits = model(features["pixel_values"])[-1]
13
  probability = torch.nn.functional.softmax(logits, dim=-1)
14
  probs = probability[0].detach().numpy()