Fix for redaction at end of text (#45)

remotes/1710274000886183304/main
mrq 2023-06-10 21:16:21 +07:00
parent 5ff00bf3bf
commit e2cd07d560
1 changed files with 1 additions and 1 deletions

@ -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])