hlydecker commited on
Commit
f52cc6a
·
verified ·
1 Parent(s): 075db70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def classify_image(image):
35
  with torch.no_grad():
36
  model.eval()
37
 
38
- inputs = feature_extractor(images=image, return_tensors="pt")
39
  outputs = model(**inputs)
40
 
41
  logits = outputs.logits
 
35
  with torch.no_grad():
36
  model.eval()
37
 
38
+ inputs = processor(images=image, return_tensors="pt")
39
  outputs = model(**inputs)
40
 
41
  logits = outputs.logits