Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ def infer():
|
|
50 |
transforms = weights.transforms()
|
51 |
|
52 |
|
53 |
-
def preprocess(
|
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)
|