Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -101,8 +101,8 @@ def inference(file, mode):
|
|
101 |
elif mode == "inpainting":
|
102 |
model = {'name':'GPEN-Inpainting-1024', 'size':1024}
|
103 |
faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
|
104 |
-
|
105 |
-
inpaint = faceinpainter.process(
|
106 |
|
107 |
cv2.imwrite(os.path.join("output.png"), inpaint)
|
108 |
return os.path.join("output.png")
|
|
|
101 |
elif mode == "inpainting":
|
102 |
model = {'name':'GPEN-Inpainting-1024', 'size':1024}
|
103 |
faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
|
104 |
+
brokenf = np.asarray(brush_stroke_mask(Image.fromarray(im)))
|
105 |
+
inpaint = faceinpainter.process(brokenf)
|
106 |
|
107 |
cv2.imwrite(os.path.join("output.png"), inpaint)
|
108 |
return os.path.join("output.png")
|