From af6931d1de362a8261c899dea871edc7c4f43b46 Mon Sep 17 00:00:00 2001 From: Gabriel Ziegler Date: Wed, 9 Feb 2022 19:18:48 -0300 Subject: [PATCH] Fix bug where min_speech_duration_ms is not checked in the last speech segment Signed-off-by: Gabriel Ziegler --- utils_vad.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/utils_vad.py b/utils_vad.py index eccf618..72e91e6 100644 --- a/utils_vad.py +++ b/utils_vad.py @@ -240,8 +240,9 @@ def get_speech_timestamps(audio: torch.Tensor, continue if current_speech: - current_speech['end'] = audio_length_samples - speeches.append(current_speech) + if (audio_length_samples - current_speech['start']) > min_speech_samples: + current_speech['end'] = audio_length_samples + speeches.append(current_speech) for i, speech in enumerate(speeches): if i == 0: