From e6ecbaa68fcbb5a5010d42fc77d000eb362a6964 Mon Sep 17 00:00:00 2001 From: Ahmad Bilal Date: Thu, 20 Jul 2023 03:20:47 +0500 Subject: [PATCH] Remove spacing --- whisperx/diarize.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/whisperx/diarize.py b/whisperx/diarize.py index b14addf..320d2a4 100644 --- a/whisperx/diarize.py +++ b/whisperx/diarize.py @@ -27,7 +27,6 @@ class DiarizationPipeline: def assign_word_speakers(diarize_df, transcript_result, fill_nearest=False): transcript_segments = transcript_result["segments"] for seg in transcript_segments: - # assign speaker to segment (if any) diarize_df['intersection'] = np.minimum(diarize_df['end'], seg['end']) - np.maximum(diarize_df['start'], seg['start']) diarize_df['union'] = np.maximum(diarize_df['end'], seg['end']) - np.minimum(diarize_df['start'], seg['start']) @@ -44,7 +43,6 @@ def assign_word_speakers(diarize_df, transcript_result, fill_nearest=False): # assign speaker to words if 'words' in seg: for word in seg['words']: - 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'])