fffiloni commited on
Commit
3a87a23
1 Parent(s): 4cbdce1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def infer():
50
  transforms = weights.transforms()
51
 
52
 
53
- def preprocess(img, img2):
54
  img1 = F.resize(img1, size=[520, 960])
55
  img2 = F.resize(img2, size=[520, 960])
56
  return transforms(img1, img2)
 
50
  transforms = weights.transforms()
51
 
52
 
53
+ def preprocess(img1, img2):
54
  img1 = F.resize(img1, size=[520, 960])
55
  img2 = F.resize(img2, size=[520, 960])
56
  return transforms(img1, img2)