adamnsandle
2023-04-28 11:48:01 +00:00
parent c3c67cdcb8
commit a9d2b591de
2 changed files with 4 additions and 5 deletions

Binary file not shown.

View File

@@ -34,7 +34,7 @@ class OnnxWrapper():
if sr != 16000 and (sr % 16000 == 0): if sr != 16000 and (sr % 16000 == 0):
step = sr // 16000 step = sr // 16000
x = x[::step] x = x[:, ::step]
sr = 16000 sr = 16000
if sr not in self.sample_rates: if sr not in self.sample_rates:
@@ -310,7 +310,6 @@ def get_speech_timestamps(audio: torch.Tensor,
triggered = False triggered = False
continue continue
if (speech_prob < neg_threshold) and triggered: if (speech_prob < neg_threshold) and triggered:
if not temp_end: if not temp_end:
temp_end = window_size_samples * i temp_end = window_size_samples * i