Update app.py

app.py

@@ -44,9 +44,10 @@ 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
+from PIL import Image
+
+def resize_image(image, size):
+    width, height = image.size
     if max(width, height) > size:
         if width > height:
             ratio = width / size
@@ -56,10 +57,10 @@ def resize_image(input_image_path, size):
             ratio = height / size
             new_width = int(width / ratio)
             new_height = size
-        resized_image = original_image.resize((new_width, new_height), Image.ANTIALIAS)
+        resized_image = image.resize((new_width, new_height), Image.ANTIALIAS)
         return resized_image
     else:
-        return original_image
+        return image
 
 
 def nogan(input_img):