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():
|