Update face_enhancement.py

face_enhancement.py

@@ -15,11 +15,11 @@ from sr_model.real_esrnet import RealESRNet
 from align_faces import warp_and_crop_face, get_reference_facial_points
 
 class FaceEnhancement(object):
-    def __init__(self, base_dir='./', size=512, out_size=None, model=None, channel_multiplier=2, narrow=1, key=None, device='cpu'):
+    def __init__(self, base_dir='./', size=512, out_size=None, model=None, channel_multiplier=2, narrow=1, key=None, device='cpu', u=False):
         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, 'realesrnet', 2, 0, device=device)
-        self.use_sr = True
+        self.use_sr = u
         self.size = size
         self.out_size = size if out_size==None else out_size
         self.threshold = 0.9