Eason Lu commited on
Commit
5ffd3bc
·
1 Parent(s): fe8b7a1

small fix on video name

Browse files

Former-commit-id: bb351fb8be9cc164db2a072fa0eb82b2dd4e6659

Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -80,7 +80,7 @@ def get_sources(args, download_path, result_path, video_name):
80
  audio_path = args.audio_file
81
  pass
82
 
83
- return audio_path, audio_file, video_path
84
 
85
  def get_srt_class(srt_file_en, result_path, video_name, audio_path, audio_file = None, whisper_model = 'base', method = "stable"):
86
  # Instead of using the script_en variable directly, we'll use script_input
@@ -249,7 +249,7 @@ def main():
249
  else:
250
  VIDEO_NAME = args.video_name
251
 
252
- audio_path, audio_file, video_path = get_sources(args, DOWNLOAD_PATH, RESULT_PATH, VIDEO_NAME)
253
 
254
  srt_file_en, srt = get_srt_class(args.srt_file, RESULT_PATH, VIDEO_NAME, audio_path, audio_file)
255
 
 
80
  audio_path = args.audio_file
81
  pass
82
 
83
+ return audio_path, audio_file, video_path, video_name
84
 
85
  def get_srt_class(srt_file_en, result_path, video_name, audio_path, audio_file = None, whisper_model = 'base', method = "stable"):
86
  # Instead of using the script_en variable directly, we'll use script_input
 
249
  else:
250
  VIDEO_NAME = args.video_name
251
 
252
+ audio_path, audio_file, video_path, VIDEO_NAME = get_sources(args, DOWNLOAD_PATH, RESULT_PATH, VIDEO_NAME)
253
 
254
  srt_file_en, srt = get_srt_class(args.srt_file, RESULT_PATH, VIDEO_NAME, audio_path, audio_file)
255