KenjieDec commited on
Commit
65f117f
·
1 Parent(s): 6a818b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -101,7 +101,7 @@ def inference(file, mode):
101
  return os.path.join("output.png")
102
  elif mode == "inpainting":
103
  im1 = cv2.imread(file, cv2.IMREAD_COLOR)
104
- im2 = imutils.resize(im1, width=1024)
105
  im3 = cv2.resize(im2, (0,0), fx=3, fy=3)
106
  model = {'name':'GPEN-Inpainting-1024', 'size':1024}
107
  faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
 
101
  return os.path.join("output.png")
102
  elif mode == "inpainting":
103
  im1 = cv2.imread(file, cv2.IMREAD_COLOR)
104
+ im2 = imutils.resize(im1, width=2048)
105
  im3 = cv2.resize(im2, (0,0), fx=3, fy=3)
106
  model = {'name':'GPEN-Inpainting-1024', 'size':1024}
107
  faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')