mrrandom123
commited on
Commit
•
d804b9f
1
Parent(s):
36709f6
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ def predict(inp):
|
|
14 |
logits = outputs.logits.cpu()
|
15 |
upsampled_logits = nn.functional.interpolate(
|
16 |
logits,
|
17 |
-
size=
|
18 |
mode="bilinear",
|
19 |
align_corners=False,
|
20 |
)
|
|
|
14 |
logits = outputs.logits.cpu()
|
15 |
upsampled_logits = nn.functional.interpolate(
|
16 |
logits,
|
17 |
+
size=inp.size[::-1],
|
18 |
mode="bilinear",
|
19 |
align_corners=False,
|
20 |
)
|