Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,12 @@ from stf_utils import STFPipeline
|
|
55 |
|
56 |
|
57 |
|
58 |
-
|
|
|
|
|
|
|
|
|
|
|
59 |
|
60 |
|
61 |
@spaces.GPU()
|
@@ -76,12 +81,7 @@ def gpu_wrapped_stf_pipeline_execute(audio_path, video_type):
|
|
76 |
tyro.extras.set_accent_color("bright_cyan")
|
77 |
args = tyro.cli(ArgumentConfig)
|
78 |
driving_video_path=gr.Video()
|
79 |
-
|
80 |
-
stf_pipeline_male = STFPipeline(
|
81 |
-
template_video_path="TEMP/Cam2_2309071202_0012_Natural_Looped.mp4",
|
82 |
-
config_path="TEMP/front_config_v3.json",
|
83 |
-
checkpoint_path="TEMP/0157.pth",
|
84 |
-
)
|
85 |
|
86 |
with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
87 |
with gr.Row():
|
|
|
55 |
|
56 |
|
57 |
|
58 |
+
stf_pipeline_female = STFPipeline()
|
59 |
+
stf_pipeline_male = STFPipeline(
|
60 |
+
template_video_path="TEMP/Cam2_2309071202_0012_Natural_Looped.mp4",
|
61 |
+
config_path="TEMP/front_config_v3.json",
|
62 |
+
checkpoint_path="TEMP/0157.pth",
|
63 |
+
)
|
64 |
|
65 |
|
66 |
@spaces.GPU()
|
|
|
81 |
tyro.extras.set_accent_color("bright_cyan")
|
82 |
args = tyro.cli(ArgumentConfig)
|
83 |
driving_video_path=gr.Video()
|
84 |
+
|
|
|
|
|
|
|
|
|
|
|
85 |
|
86 |
with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
87 |
with gr.Row():
|