Update app.py

app.py

@@ -18,21 +18,20 @@ from face_colorization import FaceColorization
   
 
 def inference(file, mode):
-
+    im = cv2.imread(file, cv2.IMREAD_COLOR)
+    im = cv2.resize(im, (0,0), fx=2, fy=2)
+    faceenhancer = FaceEnhancement(size=512, model='GPEN-512', channel_multiplier=2, device='cpu')
+    img, orig_faces, enhanced_faces = faceenhancer.process(im)
+    
     if mode == "enhance":
-        model = {'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=model['name'], channel_multiplier=2, device='cpu')
-        img, orig_faces, enhanced_faces = faceenhancer.process(im)
+        # model = {'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=model['name'], channel_multiplier=2, device='cpu')
+        # img, orig_faces, enhanced_faces = faceenhancer.process(im)
         cv2.imwrite(os.path.join("output.png"), img)
         return os.path.join("output.png")
     else:
-        im = cv2.imread(file, cv2.IMREAD_COLOR)
-        im = cv2.resize(im, (0,0), fx=2, fy=2)
-        faceenhancer = FaceEnhancement(size=512, model='GPEN-512', channel_multiplier=2, device='cpu')
-        img, orig_faces, enhanced_faces = faceenhancer.process(im)
-        
         model = {'name':'GPEN-1024-Color', 'size':1024}
         grayf = cv2.imread("enhanced.png", cv2.IMREAD_GRAYSCALE)
         grayf = cv2.cvtColor(grayf, cv2.COLOR_GRAY2BGR) # channel: 1->3