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

Update alignment.py

Browse Source
This commit is contained in:
awerks
2023-08-17 14:56:54 +02:00
committed by GitHub
parent abbb66b58e
commit 6cb7267dc2
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
whisperx/alignment.py
View File

@ -99,7 +99,8 @@ def align(
interpolate_method: str = "nearest",
return_char_alignments: bool = False,
print_progress = False,
combined_progress = False
combined_progress = False,
total_segments = 0
) -> AlignedTranscriptionResult:
"""
Align phoneme recognition predictions to known transcription.
@ -118,9 +119,6 @@ def align(
model_lang = align_model_metadata["language"]
model_type = align_model_metadata["type"]
total_segments = len(list(transcript))
transcript = iter(transcript)
# 1. Preprocess to keep only characters in dictionary
for sdx, segment in enumerate(transcript):
# strip spaces at beginning / end, but keep track of the amount.