diff --git a/rust-whisper-baked/src/main.rs b/rust-whisper-baked/src/main.rs index dd96f1d..cf9cd9f 100644 --- a/rust-whisper-baked/src/main.rs +++ b/rust-whisper-baked/src/main.rs @@ -100,8 +100,8 @@ impl Destutterer { let next_words = Words::from_string(next.clone()); let mut n = self.prevs.len().clamp(0, next_words.len()); while n > 0 { - let prev_s = (self.prevs.len()-n..self.prevs.len()).map(|i| self.prevs.comparable_string(i)).collect().join(" "); - let next_s = (0-n).map(|i| next_words.comparable_string(i)).collect().join(" "); + let prev_s, _ = self.prevs.last_n_comparable_to_string(n); + let next_s, _ = next_words.first_n_comparable_to_string(n); eprintln!("prevs => '{}'", &prev_s); eprintln!("nexts => '{}'", &next_s); if prev_s == next_s {