yerang commited on
Commit
2b756ea
1 Parent(s): f55c77d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -142,14 +142,16 @@ def run_end_to_end(image_path, text, voice, input_video, flag_relative, flag_do_
142
  # args=args
143
  # )
144
 
 
 
145
 
146
- if not male:
147
- stf_pipeline = STFPipeline()
148
- else:
149
- stf_pipeline = STFPipeline(template_video_path="/home/user/app/stf/TEMP/Cam2_2309071202_0012_Natural_Looped.mp4",
150
- config_path="/home/user/app/stf/TEMP/front_config_v3.json",
151
- checkpoint_path="/home/user/app/stf/TEMP/0157.pth",
152
- )
153
 
154
  if input_video is None:
155
  #audio_path = elevenlabs_pipeline.generate_voice(text, voice)
 
142
  # args=args
143
  # )
144
 
145
+
146
+ if input_video is None:
147
 
148
+ if not male:
149
+ stf_pipeline = STFPipeline()
150
+ else:
151
+ stf_pipeline = STFPipeline(template_video_path="/home/user/app/stf/TEMP/Cam2_2309071202_0012_Natural_Looped.mp4",
152
+ config_path="/home/user/app/stf/TEMP/front_config_v3.json",
153
+ checkpoint_path="/home/user/app/stf/TEMP/0157.pth",
154
+ )
155
 
156
  if input_video is None:
157
  #audio_path = elevenlabs_pipeline.generate_voice(text, voice)