Spaces:
Sleeping
Sleeping
Alp İpekçiler
commited on
Commit
·
db44cf4
1
Parent(s):
888d9dd
Fix: Remove padding and use images parameter correctly for ConvNext processor
Browse files
app.py
CHANGED
|
@@ -94,7 +94,7 @@ def predict_pet():
|
|
| 94 |
|
| 95 |
# Make prediction
|
| 96 |
import torch
|
| 97 |
-
inputs = image_processor(pil_image, return_tensors="pt"
|
| 98 |
|
| 99 |
device = next(model.parameters()).device
|
| 100 |
inputs = {k: v.to(device) for k, v in inputs.items()}
|
|
|
|
| 94 |
|
| 95 |
# Make prediction
|
| 96 |
import torch
|
| 97 |
+
inputs = image_processor(images=pil_image, return_tensors="pt")
|
| 98 |
|
| 99 |
device = next(model.parameters()).device
|
| 100 |
inputs = {k: v.to(device) for k, v in inputs.items()}
|