demo_contextual_paraformer.py 497 B

123456789101112
  1. from funasr_onnx import ContextualParaformer
  2. from pathlib import Path
  3. model_dir = "./export/damo/speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404"
  4. model = ContextualParaformer(model_dir, batch_size=1)
  5. wav_path = ['{}/.cache/modelscope/hub/damo/speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404/example/asr_example.wav'.format(Path.home())]
  6. hotwords = '随机热词 各种热词 魔搭 阿里巴巴 仏'
  7. result = model(wav_path, hotwords)
  8. print(result)