From e2cd07d5602742ec4068195532afa2ec4d56bebe Mon Sep 17 00:00:00 2001 From: mrq Date: Sat, 10 Jun 2023 21:16:21 +0000 Subject: [PATCH] Fix for redaction at end of text (#45) --- tortoise/utils/wav2vec_alignment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tortoise/utils/wav2vec_alignment.py b/tortoise/utils/wav2vec_alignment.py index 35c9bb5..c1647af 100755 --- a/tortoise/utils/wav2vec_alignment.py +++ b/tortoise/utils/wav2vec_alignment.py @@ -144,7 +144,7 @@ class Wav2VecAlignment: non_redacted_intervals = [] last_point = 0 for i in range(len(fully_split)): - if i % 2 == 0: + if i % 2 == 0 and fully_split[i] != "": # Check for empty string fixes index error end_interval = max(0, last_point + len(fully_split[i]) - 1) non_redacted_intervals.append((last_point, end_interval)) last_point += len(fully_split[i])