KenjieDec commited on
Commit
0e7c766
·
1 Parent(s): 079134b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -132,11 +132,11 @@ 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 = 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)))
139
- inpaint = faceinpainter.process(im2)
140
 
141
  cv2.imwrite(os.path.join("output.png"), inpaint)
142
  return os.path.join("output.png")
 
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
+ im3 = np.asarray(brush_stroke_mask(Image.fromarray(im2)))
139
+ inpaint = faceinpainter.process(im3)
140
 
141
  cv2.imwrite(os.path.join("output.png"), inpaint)
142
  return os.path.join("output.png")