app.py

@@ -21,14 +21,14 @@ def inference(file, mode):
     if mode == "enhance":
         model = {'name':'GPEN-512', 'size':512}
         im = cv2.imread(file, cv2.IMREAD_COLOR) 
-        faceenhancer = FaceEnhancement(size=model['size'], model=model['name'], channel_multiplier=2)
+        faceenhancer = FaceEnhancement(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
         img, orig_faces, enhanced_faces = faceenhancer.process(im)
         return enhanced_faces[0][:,:,::-1]
     else:
         model = {'name':'GPEN-1024-Color', 'size':1024}
         grayf = cv2.imread(file, cv2.IMREAD_GRAYSCALE)
         grayf = cv2.cvtColor(grayf, cv2.COLOR_GRAY2BGR) # channel: 1->3
-        facecolorizer = FaceColorization(size=model['size'], model=model['name'])
+        facecolorizer = FaceColorization(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
         colorf = facecolorizer.process(grayf)
 
         colorf = cv2.resize(colorf, (grayf.shape[1], grayf.shape[0]))