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

added if clause for checking

Browse Source
This commit is contained in:
arnavmehta7
2023-02-10 14:48:51 +05:30
parent d687cf3358
commit 2e307814dd
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
whisperx/alignment.py
View File

@ -389,10 +389,11 @@ def align(
wseg["segment-text-end"].fillna(len(text)-1, inplace=True) wseg["segment-text-end"].fillna(len(text)-1, inplace=True)
cseg = char_segments_arr.loc[seg_idx].loc[sub_start:sub_end-1] cseg = char_segments_arr.loc[seg_idx].loc[sub_start:sub_end-1]
cseg['segment-text-start'] = cseg['level_1'] # fixes bug for single segment in transcript
cseg['segment-text-end'] = cseg['level_1'] + 1 cseg['segment-text-start'] = cseg['level_1'] if 'level_1' in cseg else 0
del cseg['level_1'] cseg['segment-text-end'] = cseg['level_1'] + 1 if 'level_1' in cseg else 1
del cseg['level_0'] if 'level_1' in cseg: del cseg['level_1']
if 'level_0' in cseg: del cseg['level_0']
cseg.reset_index(inplace=True) cseg.reset_index(inplace=True)
aligned_segments.append( aligned_segments.append(
{ {