KenjieDec commited on
Commit
c9f5c91
1 Parent(s): 30f3b80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def inference(file, mode):
38
  cv2.imwrite(os.path.join("output.png"), colorf)
39
  return os.path.join("output.png")
40
  else:
41
- faceenhancer = FaceEnhancement(size=512, model='GPEN-512', channel_multiplier=2, device='cpu', u=False)
42
  img, orig_faces, enhanced_faces = faceenhancer.process(im)
43
  cv2.imwrite(os.path.join("output.png"), img)
44
  return os.path.join("output.png")
 
38
  cv2.imwrite(os.path.join("output.png"), colorf)
39
  return os.path.join("output.png")
40
  else:
41
+ faceenhancer = FaceEnhancement(size=512, model='GPEN-512', channel_multiplier=2, device='cpu', u=True)
42
  img, orig_faces, enhanced_faces = faceenhancer.process(im)
43
  cv2.imwrite(os.path.join("output.png"), img)
44
  return os.path.join("output.png")