From 1bb4839b0f8d5d10a19ac64aae91623d8ad96db0 Mon Sep 17 00:00:00 2001 From: awerks Date: Wed, 16 Aug 2023 16:13:28 +0200 Subject: [PATCH] Update alignment.py --- whisperx/alignment.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/whisperx/alignment.py b/whisperx/alignment.py index 0628f42..bda322c 100644 --- a/whisperx/alignment.py +++ b/whisperx/alignment.py @@ -162,9 +162,15 @@ def align( segment["sentence_spans"] = sentence_spans aligned_segments: List[SingleAlignedSegment] = [] - + total_segments = len(list(transcript)) + transcript = iter(transcript) + # 2. Get prediction matrix from alignment model & align for sdx, segment in enumerate(transcript): + + percent_complete = ((sdx + 1) / total_segments) * 100 + print(f"Progress: {percent_complete:.2f}%...") + t1 = segment["start"] t2 = segment["end"] text = segment["text"]