Update app.py

app.py

@@ -91,7 +91,7 @@ def inference(file, mode):
     elif mode == "colorize":
         model = {'name':'GPEN-1024-Color', 'size':1024}
         grayf = cv2.imread("e.png", cv2.IMREAD_GRAYSCALE)
-        grayf = cv2.cvtColor(grayf, cv2.COLOR_GRAY2BGR)
+        grayf = cv2.cvtColor(grayf, cv2.COLOR_GRAY2BGR) # channel: 1->3
         facecolorizer = FaceColorization(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
         colorf = facecolorizer.process(grayf)
 
@@ -101,8 +101,8 @@ 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')
-        brokenf = np.asarray(brush_stroke_mask(Image.fromarray(im)))
-        inpaint = faceinpainter.process(brokenf)
+        im = np.asarray(brush_stroke_mask(Image.fromarray(im)))
+        inpaint = faceinpainter.process(im)
 
         cv2.imwrite(os.path.join("output.png"), inpaint)
         return os.path.join("output.png")