sshi commited on
Commit
3efc2d7
1 Parent(s): d78a0b4

App bug fix.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -118,8 +118,8 @@ def plot_results(pil_img, prob, boxes):
118
 
119
  def generate_preds(processor, model, image):
120
  inputs = processor(images=image, return_tensors="pt").to(device)
121
- pixel_values = inputs.pixel_values.unsqueeze(0)
122
- preds = model(pixel_values=pixel_values)
123
  return preds
124
 
125
 
 
118
 
119
  def generate_preds(processor, model, image):
120
  inputs = processor(images=image, return_tensors="pt").to(device)
121
+ # pixel_values = inputs.pixel_values.unsqueeze(0)
122
+ preds = model(pixel_values=inputs.pixel_values)
123
  return preds
124
 
125