Update app.py

app.py

@@ -44,7 +44,26 @@ def denormalize(x, mean=(0., 0., 0.), std=(1.0, 1.0, 1.0)):
 
     return x 
 
+def resize_image(input_image_path, size):
+    original_image = Image.open(input_image_path)
+    width, height = original_image.size
+    if max(width, height) > size:
+        if width > height:
+            ratio = width / size
+            new_height = int(height / ratio)
+            new_width = size
+        else:
+            ratio = height / size
+            new_width = int(width / ratio)
+            new_height = size
+        resized_image = original_image.resize((new_width, new_height), Image.ANTIALIAS)
+        return resized_image
+    else:
+        return original_image
+
+
 def nogan(input_img):
+    input_img = resize_image(input_img, 1024)
     i = np.asarray(input_img)
     i = i.astype("float32")
     i = np.transpose(i, (2, 0, 1))