Default to english

This commit is contained in:
Marty Oehme 2024-03-14 08:43:16 +01:00
parent 56d7406a31
commit 5d93fb7ad3
Signed by: Marty
GPG key ID: EDBF2ED917B2EF6A

View file

@ -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(