Update app.py

app.py

@@ -25,7 +25,6 @@ from face_inpainting import FaceInpainting
 
 
 def inference(file, mode):
-
     
     im_orig = cv2.imread(file, cv2.IMREAD_COLOR)
     im = cv2.resize(im_orig, (0,0), fx=2, fy=2)
@@ -47,8 +46,7 @@ def inference(file, mode):
     elif mode == "inpainting":
         model = {'name':'GPEN-Inpainting-1024', 'size':1024}
         faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
-        im = np.asarray(brush_stroke_mask(Image.fromarray(im_orig)))
-        inpaint = faceinpainter.process(im)
+        inpaint = faceinpainter.process(im_orig)
         cv2.imwrite(os.path.join("output.png"), inpaint)
         return os.path.join("output.png")
     elif mode == "selfie":