chore: refine the doc and code

.gitattributes

@@ -47,3 +47,5 @@ pretrained_weights/liveportrait/base_models/warping_module.pth filter=lfs diff=l
 pretrained_weights/insightface/models/buffalo_l/2d106det.onnx filter=lfs diff=lfs merge=lfs -text
 pretrained_weights/insightface/models/buffalo_l/det_10g.onnx filter=lfs diff=lfs merge=lfs -text
 pretrained_weights/liveportrait/landmark.onnx filter=lfs diff=lfs merge=lfs -text
+assets/examples/driving/d14.mp4 filter=lfs diff=lfs merge=lfs -text
+assets/examples/source/s12.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -46,14 +46,14 @@ def gpu_wrapped_execute_image(*args, **kwargs):
 
 def is_square_video(video_path):
     video = cv2.VideoCapture(video_path)
-    
+
     width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
-    
+
     video.release()
     if width != height:
         raise gr.Error("Error: the video does not have a square aspect ratio. We currently only support square videos")
-    
+
     return gr.update(visible=True)
 
 # assets
@@ -91,6 +91,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     [osp.join(example_portrait_dir, "s10.jpg")],
                     [osp.join(example_portrait_dir, "s5.jpg")],
                     [osp.join(example_portrait_dir, "s7.jpg")],
+                    [osp.join(example_portrait_dir, "s12.jpg")],
                 ],
                 inputs=[image_input],
                 cache_examples=False,
@@ -100,6 +101,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             gr.Examples(
                 examples=[
                     [osp.join(example_video_dir, "d0.mp4")],
+                    [osp.join(example_video_dir, "d14.mp4")],
                     [osp.join(example_video_dir, "d5.mp4")],
                     [osp.join(example_video_dir, "d6.mp4")],
                     [osp.join(example_video_dir, "d7.mp4")],
@@ -128,7 +130,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 output_video_concat.render()
     with gr.Row():
         # Examples
-        gr.Markdown("## You could choose the examples below ⬇️")
+        gr.Markdown("## You could also choose the examples below by one click ⬇️")
     with gr.Row():
         gr.Examples(
             examples=data_examples,

assets/examples/driving/d14.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:465e72fbf26bf4ed46d1adf7aab8a7344aac54a2f92c4d82a1d53127f0170472
+size 891025

assets/gradio_description_retargeting.md

@@ -1 +1,4 @@
-<span style="font-size: 1.2em;">🔥 To change the eyes and lip open ratio of the source portrait, please drag the sliders and then click the <strong>🚗 Retargeting</strong> button. The result would be shown in the blocks. You can try running it multiple times. <strong>😊 Set both ratios to 0.8 to see what's going on!</strong> </span>
+<br>
+
+## Retargeting
+<span style="font-size: 1.2em;">🔥 To edit the eyes and lip open ratio of the source portrait, drag the sliders and click the <strong>🚗 Retargeting</strong> button. You can try running it multiple times. <strong>😊 Set both ratios to 0.8 to see what's going on!</strong> </span>

src/config/argument_config.py

@@ -8,6 +8,7 @@ import os.path as osp
 from dataclasses import dataclass
 import tyro
 from typing_extensions import Annotated
+from typing import Optional
 from .base_config import PrintableConfig, make_abs_path
 
 
@@ -41,4 +42,4 @@ class ArgumentConfig(PrintableConfig):
     ########## gradio arguments ##########
     server_port: Annotated[int, tyro.conf.arg(aliases=["-p"])]  = 7860
     share: bool = False
-    server_name: str = None # one can set "0.0.0.0" on local
+    server_name: Optional[str] = None  # one can set "0.0.0.0" on local

src/gradio_pipeline.py

@@ -10,7 +10,6 @@ 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
@@ -111,4 +110,3 @@ class GradioPipeline(LivePortraitPipeline):
         else:
             # when press the clear button, go here
             raise gr.Error("The retargeting input hasn't been prepared yet 💥!", duration=5)
-