diff --git a/verbanote/rp_handler.py b/verbanote/rp_handler.py index b4ca710..add40e4 100644 --- a/verbanote/rp_handler.py +++ b/verbanote/rp_handler.py @@ -16,6 +16,7 @@ access_token: str = os.environ.get("VERBANOTE_HF_TOKEN", "") loaders.prep() diarize_pipeline = loaders.diarization(access_token) +# TODO download model into Docker image beforehand whisper_model = loaders.whispermodel() @@ -42,7 +43,7 @@ def handler(job): model=whisper_model, diarized_groups=diarized_groups, files_path=output_path, - lang=lang or "fr", + lang=lang or "en", word_timestamps=word_timestamps or True, ) transcription: process.TxtTranscription = process.output_txt(