FunASR/runtime/python/onnxruntime/demo_seaco_paraformer.py

13 lines
389 B
Python
Raw Normal View History

2024-05-18 15:50:56 +08:00
from funasr_onnx import SeacoParaformer
from pathlib import Path
model_dir = "iic/speech_seaco_paraformer_large_asr_nat-zh-cn-16k-common-vocab8404-pytorch"
model = SeacoParaformer(model_dir, batch_size=1)
wav_path = ["{}/.cache/modelscope/hub/{}/example/asr_example.wav".format(Path.home(), model_dir)]
hotwords = "你的热词 魔搭"
result = model(wav_path, hotwords)
print(result)