Update src/gradio_pipeline.py

src/gradio_pipeline.py

@@ -10,6 +10,7 @@ from .utils.io import load_img_online
 from .utils.rprint import rlog as log
 from .utils.crop import prepare_paste_back, paste_back
 # from .utils.camera import get_rotation_matrix
+from .utils.retargeting_utils import calc_eye_close_ratio, calc_lip_close_ratio
 
 def update_args(args, user_args):
     """update the args according to user inputs
@@ -122,7 +123,8 @@ class GradioPipeline(LivePortraitPipeline):
             ######## process source portrait ########
             img_rgb = load_img_online(input_image, mode='rgb', max_dim=1280, n=16)
             log(f"Load source image from {input_image}.")
-            crop_info = self.cropper.crop_source_image(img_rgb, self.cropper.crop_cfg)
+            #crop_info = self.cropper.crop_source_image(img_rgb, self.cropper.crop_cfg)
+            crop_info = self.cropper.crop_single_image(img_rgb)
             if crop_info is None:
                 raise gr.Error("Source portrait NO face detected", duration=2)
             source_eye_ratio = calc_eye_close_ratio(crop_info['lmk_crop'][None])