datasci/verbanote-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Default to english

Browse source
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
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
verbanote/rp_handler.py
View file

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