From c69cb6c9c01ad2c3483c46df2911e13061e14351 Mon Sep 17 00:00:00 2001 From: Saenyakorn Siangsanoh Date: Wed, 28 Dec 2022 14:40:24 +0700 Subject: [PATCH] fix progress logic --- utils_vad.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/utils_vad.py b/utils_vad.py index 09aff37..603f0ca 100644 --- a/utils_vad.py +++ b/utils_vad.py @@ -264,13 +264,12 @@ def get_speech_timestamps(audio: torch.Tensor, speech_prob = model(chunk, sampling_rate).item() speech_probs.append(speech_prob) # caculate progress and seng it to callback function - progress = (current_start_sample / audio_length_samples) * 100 + progress = current_start_sample + window_size_samples + if progress > audio_length_samples: + progress = audio_length_samples + progress_percent = (progress / audio_length_samples) * 100 if progress_tracking_callback: - progress_tracking_callback(progress) - - # sending 100% progress to callback function after processing with actual model - if progress_tracking_callback: - progress_tracking_callback(100) + progress_tracking_callback(progress_percent) triggered = False speeches = []