yerang commited on
Commit
0eb45e0
·
verified ·
1 Parent(s): 3b6cc24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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
- stf_pipeline_female = STFPipeline()
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():