demo.py 1.3 KB

1234567891011121314151617181920212223242526272829303132
  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="iic/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
  7. model_revision="v2.0.4",
  8. vad_model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch",
  9. vad_model_revision="v2.0.4",
  10. punc_model="iic/punc_ct-transformer_zh-cn-common-vocab272727-pytorch",
  11. punc_model_revision="v2.0.4",
  12. # spk_model="iic/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_example_zh.wav")
  16. print(res)
  17. ''' can not use currently
  18. from funasr import AutoFrontend
  19. frontend = AutoFrontend(model="damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch", model_revision="v2.0.4")
  20. fbanks = frontend(input="https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav", batch_size=2)
  21. for batch_idx, fbank_dict in enumerate(fbanks):
  22. res = model.generate(**fbank_dict)
  23. print(res)
  24. '''