From 4acb5b3abcf667a9bbf36891e6cd9fc556a7a0a4 Mon Sep 17 00:00:00 2001 From: awerks Date: Wed, 16 Aug 2023 16:11:46 +0200 Subject: [PATCH] Update asr.py --- whisperx/asr.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/whisperx/asr.py b/whisperx/asr.py index ecc2765..ba65a95 100644 --- a/whisperx/asr.py +++ b/whisperx/asr.py @@ -285,7 +285,10 @@ class FasterWhisperPipeline(Pipeline): segments: List[SingleSegment] = [] batch_size = batch_size or self._batch_size + total_segments = len(vad_segments) for idx, out in enumerate(self.__call__(data(audio, vad_segments), batch_size=batch_size, num_workers=num_workers)): + percent_complete = ((idx + 1) / total_segments) * 100 + print(f"Progress: {percent_complete:.2f}%...") text = out['text'] if batch_size in [0, 1, None]: text = text[0]