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

Rectify refernce to the word

Browse Source
This commit is contained in:
Ahmad Bilal
2023-07-20 02:54:06 +05:00
parent 30eff5a01f
commit eb712f3999
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisperx/diarize.py
View File

@ -46,7 +46,7 @@ def assign_word_speakers(diarize_df, transcript_result, fill_nearest=False):
# assign speaker to words
if 'words' in seg:
for word in seg['words']:
word = word.replace('"', "'")
word["word"] = word["word"].replace('"', "'")
if 'start' in word:
diarize_df['intersection'] = np.minimum(diarize_df['end'], word['end']) - np.maximum(diarize_df['start'], word['start'])
diarize_df['union'] = np.maximum(diarize_df['end'], word['end']) - np.minimum(diarize_df['start'], word['start'])