Spaces:
Runtime error
Runtime error
sshi
commited on
Commit
•
3efc2d7
1
Parent(s):
d78a0b4
App bug fix.
Browse files
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 |
|