Test

app.py

@@ -28,8 +28,15 @@ def inference(file, mode):
         cv2.imwrite(os.path.join("output.png"), img)
         return os.path.join("output.png")
     else:
+        emodel = {'name':'GPEN-512', 'size':512}
+        im = cv2.imread(file, cv2.IMREAD_COLOR)
+        im = cv2.resize(im, (0,0), fx=2, fy=2)
+        faceenhancer = FaceEnhancement(size=model['size'], model=emodel['name'], channel_multiplier=2, device='cpu')
+        img, orig_faces, enhanced_faces = faceenhancer.process(im)
+        cv2.imwrite(os.path.join("enhanced.png"), img)
+        
         model = {'name':'GPEN-1024-Color', 'size':1024}
-        grayf = cv2.imread(file, cv2.IMREAD_GRAYSCALE)
+        grayf = cv2.imread("enhanced.png", cv2.IMREAD_GRAYSCALE)
         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)