Commit
·
f0a6ca3
1
Parent(s):
2b4b81f
changed to cpu
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ def fn(image):
|
|
| 30 |
|
| 31 |
def process(image):
|
| 32 |
image_size = image.size
|
| 33 |
-
input_images = transform_image(image).unsqueeze(0).to("
|
| 34 |
# Prediction
|
| 35 |
with torch.no_grad():
|
| 36 |
preds = birefnet(input_images)[-1].sigmoid().cpu()
|
|
|
|
| 30 |
|
| 31 |
def process(image):
|
| 32 |
image_size = image.size
|
| 33 |
+
input_images = transform_image(image).unsqueeze(0).to("cpu")
|
| 34 |
# Prediction
|
| 35 |
with torch.no_grad():
|
| 36 |
preds = birefnet(input_images)[-1].sigmoid().cpu()
|