Update app.py

app.py

@@ -8,7 +8,7 @@ import tyro
 import gradio as gr
 import os.path as osp
 from src.utils.helper import load_description
-from src.gradio_pipeline import GradioPipeline
+from src.gradio_pipeline_stf import GradioPipeline
 from src.config.crop_config import CropConfig
 from src.config.argument_config import ArgumentConfig
 from src.config.inference_config import InferenceConfig
@@ -168,8 +168,8 @@ def run_end_to_end(image_path, text, voice, input_video, flag_relative, flag_do_
 
     #output_path, crop_output_path = gradio_pipeline.execute_video(
     output_path, crop_output_path = gpu_wrapped_execute_video(
-            input_source_image_path=image_path,
-            input_driving_video_path=driving_video_path,
+            input_image_path=image_path,
+            input_video_path=driving_video_path,
             # input_driving_video_pickle_path=None,
             flag_do_crop_input=flag_do_crop,
             # flag_remap_input=flag_remap,