Commit
•
c2e0a3f
1
Parent(s):
1ea5bb8
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,7 +45,6 @@ def fn_url(url):
|
|
| 45 |
image_size = im.size
|
| 46 |
input_images = transform_image(im).unsqueeze(0).to("cpu")
|
| 47 |
|
| 48 |
-
# Prediction
|
| 49 |
with torch.no_grad():
|
| 50 |
preds = birefnet(input_images)[-1].sigmoid().cpu()
|
| 51 |
pred = preds[0].squeeze()
|
|
|
|
| 45 |
image_size = im.size
|
| 46 |
input_images = transform_image(im).unsqueeze(0).to("cpu")
|
| 47 |
|
|
|
|
| 48 |
with torch.no_grad():
|
| 49 |
preds = birefnet(input_images)[-1].sigmoid().cpu()
|
| 50 |
pred = preds[0].squeeze()
|