diff --git a/main.py b/main.py index 21027bb..b1a3a61 100755 --- a/main.py +++ b/main.py @@ -67,6 +67,9 @@ def main(): chunk = signal[:, start_sample:end_sample] + # normalize + chunk /= chunk.abs().max() + # SpeechBrain expects [batch, time] emb = classifier.encode_batch(chunk) emb = emb.squeeze().detach().cpu().numpy()