mrrandom123 commited on
Commit
d804b9f
1 Parent(s): 36709f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=image.size[::-1],
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
  )