From 58e218d25537c536409c8d6997a51a3ad82b3007 Mon Sep 17 00:00:00 2001 From: McCloudS <64094529+McCloudS@users.noreply.github.com> Date: Sun, 22 Oct 2023 23:59:10 -0600 Subject: [PATCH] Update subgen.py --- subgen/subgen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/subgen/subgen.py b/subgen/subgen.py index 2735e01..bbac631 100644 --- a/subgen/subgen.py +++ b/subgen/subgen.py @@ -58,7 +58,7 @@ if transcribe_device == "gpu": transcribe_device = "cuda" app = Flask(__name__) -model = stable_whisper.load_faster_whisper(whisper_model, device=transcribe_device, cpu_threads=whisper_threads, num_workers=concurrent_transcriptions) +model = stable_whisper.load_faster_whisper(whisper_model, download_root="/subgen", device=transcribe_device, cpu_threads=whisper_threads, num_workers=concurrent_transcriptions) files_to_transcribe = set() subextension = '.subgen.' + whisper_model + '.' + namesublang + '.srt' print("Transcriptions are limited to running " + str(concurrent_transcriptions) + " at a time")