yuragoithf commited on
Commit
81c41e8
·
1 Parent(s): a0a4808

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def model_inference(img, prob_threshold, classes_to_show):
32
 
33
  img = Image.fromarray(img)
34
 
35
- pixel_values = feature_extractor(img, return_tensors="pt").pixel_values
36
 
37
  with torch.no_grad():
38
  outputs = model(pixel_values, output_attentions=True)
 
32
 
33
  img = Image.fromarray(img)
34
 
35
+ pixel_values = feature_extractor(img, return_tensors="pil").pixel_values
36
 
37
  with torch.no_grad():
38
  outputs = model(pixel_values, output_attentions=True)