KenjieDec commited on
Commit
079134b
·
1 Parent(s): 0775a2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -132,7 +132,7 @@ def inference(file, mode):
132
  return os.path.join("output.png")
133
  elif mode == "inpainting":
134
  im1 = cv2.imread(file, cv2.IMREAD_COLOR)
135
- im2 = image_resize(im1, width = 1024)
136
  model = {'name':'GPEN-Inpainting-1024', 'size':1024}
137
  faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
138
  im = np.asarray(brush_stroke_mask(Image.fromarray(im2)))
 
132
  return os.path.join("output.png")
133
  elif mode == "inpainting":
134
  im1 = cv2.imread(file, cv2.IMREAD_COLOR)
135
+ im2 = image_resize(im1, width = 2048)
136
  model = {'name':'GPEN-Inpainting-1024', 'size':1024}
137
  faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
138
  im = np.asarray(brush_stroke_mask(Image.fromarray(im2)))