1080

app.py

@@ -104,7 +104,7 @@ model2.eval()
 def predict(input_img, ver):
     input_img = Image.open(input_img)
     transform = transforms.Compose(
-        [transforms.Resize(512, Image.BICUBIC), transforms.ToTensor()]
+        [transforms.Resize(1080, Image.BICUBIC), transforms.ToTensor()]
     )
     input_img = transform(input_img)
     input_img = torch.unsqueeze(input_img, 0)