From fc186b28013fce8e35ab46d35d65c16a3368a114 Mon Sep 17 00:00:00 2001 From: harry Date: Thu, 28 Mar 2024 18:02:39 +0800 Subject: [PATCH] remove local_files_only=True --- app/services/subtitle.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/app/services/subtitle.py b/app/services/subtitle.py index df929f4..5ea8028 100644 --- a/app/services/subtitle.py +++ b/app/services/subtitle.py @@ -20,8 +20,7 @@ def create(audio_file, subtitle_file: str = ""): logger.info(f"loading model: {model_size}, device: {device}, compute_type: {compute_type}") model = WhisperModel(model_size_or_path=model_size, device=device, - compute_type=compute_type, - local_files_only=True) + compute_type=compute_type) logger.info(f"start, output file: {subtitle_file}") if not subtitle_file: @@ -160,6 +159,7 @@ if __name__ == "__main__": task_id = "c12fd1e6-4b0a-4d65-a075-c87abe35a072" task_dir = utils.task_dir(task_id) subtitle_file = f"{task_dir}/subtitle.srt" + audio_file = f"{task_dir}/audio.mp3" subtitles = file_to_subtitles(subtitle_file) print(subtitles) @@ -171,3 +171,6 @@ if __name__ == "__main__": script = s.get("script") correct(subtitle_file, script) + + subtitle_file = f"{task_dir}/subtitle-test.srt" + create(audio_file, subtitle_file)