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

fix(diarize): key error on empty track

Browse Source
This commit is contained in:
Marco Vela
2023-10-10 14:50:41 -05:00
parent 051047bb25
commit a5356509b6
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
whisperx/diarize.py
View File

@ -26,10 +26,9 @@ class DiarizationPipeline:
'sample_rate': SAMPLE_RATE 'sample_rate': SAMPLE_RATE
} }
segments = self.model(audio_data, min_speakers=min_speakers, max_speakers=max_speakers) segments = self.model(audio_data, min_speakers=min_speakers, max_speakers=max_speakers)
diarize_df = pd.DataFrame(segments.itertracks(yield_label=True)) diarize_df = pd.DataFrame(segments.itertracks(yield_label=True), columns=['segment', 'label', 'speaker'])
diarize_df['start'] = diarize_df[0].apply(lambda x: x.start) diarize_df['start'] = diarize_df['segment'].apply(lambda x: x.start)
diarize_df['end'] = diarize_df[0].apply(lambda x: x.end) diarize_df['end'] = diarize_df['segment'].apply(lambda x: x.end)
diarize_df.rename(columns={2: "speaker"}, inplace=True)
return diarize_df return diarize_df