Update app.py

app.py

@@ -26,6 +26,7 @@ def xdog(im, gamma=0.98, phi=200, eps=-0.1, k=1.6, sigma=1):
 
 def swin_model(img):
     h, w = img.shape[0], img.shape[1]
+    img = cv2.resize(img,(512,512))
     img = np.transpose(img[:, :, [2, 1, 0]], (2, 0, 1)) / 255.0
     img = img[None, ...].astype(np.float32)
     mod_pad_h, mod_pad_w = 0, 0
@@ -42,6 +43,7 @@ def swin_model(img):
     _, _, a, b = output.shape
     output = output[:, :, 0:a - mod_pad_h, 0:b - mod_pad_w]
     output = np.squeeze(output).clip(0, 1)
+    output = cv2.resize(output,(512,512))
     return output