Spaces:
Sleeping
Sleeping
Eason Lu
commited on
Commit
·
5ffd3bc
1
Parent(s):
fe8b7a1
small fix on video name
Browse filesFormer-commit-id: bb351fb8be9cc164db2a072fa0eb82b2dd4e6659
- 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 |
|