Update app.py

app.py

@@ -19,6 +19,7 @@ import cv2
 import glob
 import time
 import math
+import imutils
 import argparse
 import numpy as np
 from PIL import Image, ImageDraw
@@ -99,6 +100,9 @@ def inference(file, mode):
         cv2.imwrite(os.path.join("output.png"), colorf)
         return os.path.join("output.png")
     elif mode == "inpainting":
+        im = cv2.imread(file, cv2.IMREAD_COLOR)
+        im = imutils.resize(im, width=1024)
+        im = cv2.resize(im, (0,0), fx=3, fy=3)
         model = {'name':'GPEN-Inpainting-1024', 'size':1024}
         faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
         im = np.asarray(brush_stroke_mask(Image.fromarray(im)))