chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/m-bain/whisperX into main

Browse Source 
Conflicts:
	whisperx/transcribe.py
	whisperx/utils.py
This commit is contained in:
Max Bain
2023-01-24 15:38:05 +00:00
parent d395c21b83 4569cb982a
commit d1600e5b0f
5 changed files with 269 additions and 264 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisperx/transcribe.py
View File

@ -33,6 +33,7 @@ DEFAULT_ALIGN_MODELS_HF = {
"zh": "jonatasgrosman/wav2vec2-large-xlsr-53-chinese-zh-cn",
"nl": "jonatasgrosman/wav2vec2-large-xlsr-53-dutch",
"uk": "Yehor/wav2vec2-xls-r-300m-uk-with-small-lm",
"pt": "jonatasgrosman/wav2vec2-large-xlsr-53-portuguese",
}
@ -920,6 +921,5 @@ def cli():
with open(os.path.join(output_dir, audio_basename + ".char.ass"), "w", encoding="utf-8") as ass:
write_ass(result_aligned["segments"], file=ass, resolution="char")
if __name__ == "__main__":
cli()