diff --git a/utils_vad.py b/utils_vad.py index b390cf6..80db3c9 100644 --- a/utils_vad.py +++ b/utils_vad.py @@ -252,8 +252,8 @@ def get_speech_timestamps(audio: torch.Tensor, speech['end'] += int(silence_duration // 2) speeches[i+1]['start'] = int(max(0, speeches[i+1]['start'] - silence_duration // 2)) else: - speech['end'] += int(speech_pad_samples) - speeches[i+1]['start'] -= int(speech_pad_samples) + speech['end'] = int(min(audio_length_samples, speech['end'] + speech_pad_samples)) + speeches[i+1]['start'] = int(max(0, speeches[i+1]['start'] - speech_pad_samples)) else: speech['end'] = int(min(audio_length_samples, speech['end'] + speech_pad_samples))