游雁
2024-03-18 cbe2ea7e07cbf364827bd89cefc42b3f643ea3be
runtime/python/onnxruntime/demo_contextual_paraformer.py
@@ -1,12 +1,12 @@
from funasr_onnx import ContextualParaformer
from pathlib import Path
model_dir = "../export/damo/speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404" # your export dir
model_dir = "damo/speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404"
model = ContextualParaformer(model_dir, batch_size=1)
wav_path = ['{}/.cache/modelscope/hub/damo/speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404/example/asr_example.wav'.format(Path.home())]
wav_path = ['{}/.cache/modelscope/hub/{}/example/asr_example.wav'.format(Path.home(), model_dir)]
hotwords = '随机热词 各种热词 魔搭 阿里巴巴 仏'
hotwords = '你的热词 魔搭'
result = model(wav_path, hotwords)
print(result)