yerang commited on
Commit
7aa9d59
·
verified ·
1 Parent(s): 73776b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -161,8 +161,7 @@ def run_end_to_end(image_path, text, voice, input_video, flag_relative, flag_do_
161
  driving_video_path = gpu_wrapped_stf_pipeline_execute(audio_path)
162
  else:
163
  driving_video_path = input_video
164
- os.makedirs("animations",exist_ok=True)
165
- audio_path = osp.join("animations", Path(input_video).stem+".wav")
166
  extract_audio(driving_video_path, audio_path)
167
 
168
 
 
161
  driving_video_path = gpu_wrapped_stf_pipeline_execute(audio_path)
162
  else:
163
  driving_video_path = input_video
164
+ audio_path = driving_video_path.rsplit(".", 1)[0] + ".wav"
 
165
  extract_audio(driving_video_path, audio_path)
166
 
167