KenjieDec commited on
Commit
6a818b8
·
1 Parent(s): 9ba8ea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -100,13 +100,13 @@ def inference(file, mode):
100
  cv2.imwrite(os.path.join("output.png"), colorf)
101
  return os.path.join("output.png")
102
  elif mode == "inpainting":
103
- im = cv2.imread(file, cv2.IMREAD_COLOR)
104
- im = imutils.resize(im, width=1024)
105
- im = cv2.resize(im, (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')
108
- im = np.asarray(brush_stroke_mask(Image.fromarray(im)))
109
- inpaint = faceinpainter.process(im)
110
 
111
  cv2.imwrite(os.path.join("output.png"), inpaint)
112
  return os.path.join("output.png")
 
100
  cv2.imwrite(os.path.join("output.png"), colorf)
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')
108
+ im = np.asarray(brush_stroke_mask(Image.fromarray(im3)))
109
+ inpaint = faceinpainter.process(im3)
110
 
111
  cv2.imwrite(os.path.join("output.png"), inpaint)
112
  return os.path.join("output.png")