Update face_enhancement.py

face_enhancement.py

@@ -18,7 +18,7 @@ class FaceEnhancement(object):
     def __init__(self, base_dir='./', size=512, out_size=None, model=None, channel_multiplier=2, narrow=1, key=None, device='cpu'):
         self.facedetector = RetinaFaceDetection(base_dir, device)
         self.facegan = FaceGAN(base_dir, size, out_size, model, channel_multiplier, narrow, key, device=device)
-        self.srmodel = RealESRNet(base_dir, args.sr_model, args.sr_scale, args.tile_size, device=device)
+        self.srmodel = RealESRNet(base_dir, 'realesrnet', 2, 0, device=device)
         self.size = size
         self.out_size = size if out_size==None else out_size
         self.threshold = 0.9