fix demo onnx textnorm

This commit is contained in:
Zhanzhao (Deo) Liang 2024-12-25 16:59:13 +08:00 committed by GitHub
parent 530b2e3ce5
commit 4c6c6c61f3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -15,5 +15,5 @@ model = SenseVoiceSmall(model_dir, batch_size=10, quantize=True)
# inference
wav_or_scp = ["{}/.cache/modelscope/hub/{}/example/en.mp3".format(Path.home(), model_dir)]
res = model(wav_or_scp, language="auto", use_itn=True)
res = model(wav_or_scp, language="auto", textnorm="withitn")
print([rich_transcription_postprocess(i) for i in res])