demo.py 958 B

12345678910111213141516171819
  1. #!/usr/bin/env python3
  2. # -*- encoding: utf-8 -*-
  3. # Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
  4. # MIT License (https://opensource.org/licenses/MIT)
  5. from funasr import AutoModel
  6. model = AutoModel(model="damo/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
  7. model_revision="v2.0.2",
  8. vad_model="damo/speech_fsmn_vad_zh-cn-16k-common-pytorch",
  9. vad_model_revision="v2.0.2",
  10. punc_model="damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch",
  11. punc_model_revision="v2.0.2",
  12. spk_model="damo/speech_campplus_sv_zh-cn_16k-common",
  13. spk_model_revision="v2.0.2",
  14. )
  15. res = model.generate(input="https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_vad_punc_example.wav", batch_size_s=300, batch_size_threshold_s=60)
  16. print(res)