Update src/gradio_pipeline.py

src/gradio_pipeline.py

@@ -34,7 +34,13 @@ class GradioPipeline(LivePortraitPipeline):
         flag_relative_input,
         flag_do_crop_input,
         flag_remap_input,
+        flag_closed_lip_input
     ):
+
+        # Add for closed lip
+        if flag_closed_lip_input:
+            self.init_retargeting_image(self, retargeting_source_scale=2.5, input_image = input_image_path)
+        
         """ for video driven potrait animation
         """
         if input_image_path is not None and input_video_path is not None:
@@ -44,6 +50,7 @@ class GradioPipeline(LivePortraitPipeline):
                 'flag_relative': flag_relative_input,
                 'flag_do_crop': flag_do_crop_input,
                 'flag_pasteback': flag_remap_input,
+                
             }
             # update config from user input
             self.args = update_args(self.args, args_user)