From 94de39dde2e616a01683c518023d0fab72b4e103 Mon Sep 17 00:00:00 2001 From: 游雁 <zhifu.gzf@alibaba-inc.com> Date: 星期一, 19 二月 2024 22:21:50 +0800 Subject: [PATCH] aishell example --- examples/industrial_data_pretraining/seaco_paraformer/demo.py | 41 +++++++++++++++++++++++++++++++++++++---- 1 files changed, 37 insertions(+), 4 deletions(-) diff --git a/examples/industrial_data_pretraining/seaco_paraformer/demo.py b/examples/industrial_data_pretraining/seaco_paraformer/demo.py index 2e6d203..804acdd 100644 --- a/examples/industrial_data_pretraining/seaco_paraformer/demo.py +++ b/examples/industrial_data_pretraining/seaco_paraformer/demo.py @@ -5,8 +5,41 @@ from funasr import AutoModel -model = AutoModel(model="../modelscope_models/speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404") +model = AutoModel(model="iic/speech_seaco_paraformer_large_asr_nat-zh-cn-16k-common-vocab8404-pytorch", + model_revision="v2.0.4", + vad_model="damo/speech_fsmn_vad_zh-cn-16k-common-pytorch", + vad_model_revision="v2.0.4", + punc_model="damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch", + punc_model_revision="v2.0.4", + spk_model="damo/speech_campplus_sv_zh-cn_16k-common", + spk_model_revision="v2.0.2", + ) -res = model(input="../modelscope_models/speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404/example/asr_example.wav", - hotword='杈炬懇闄� 榄旀惌') -print(res) \ No newline at end of file + +# example1 +res = model.generate(input="https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav", + hotword='杈炬懇闄� 榄旀惌', + # preset_spk_num=2, + # sentence_timestamp=True, # return sentence level information when spk_model is not given + ) +print(res) + + +''' +# tensor or numpy as input +# example2 +import torchaudio +import os +wav_file = os.path.join(model.model_path, "example/asr_example.wav") +input_tensor, sample_rate = torchaudio.load(wav_file) +input_tensor = input_tensor.mean(0) +res = model.generate(input=[input_tensor], batch_size_s=300, is_final=True) + + +# example3 +import soundfile + +wav_file = os.path.join(model.model_path, "example/asr_example.wav") +speech, sample_rate = soundfile.read(wav_file) +res = model.generate(input=[speech], batch_size_s=300, is_final=True) +''' \ No newline at end of file -- Gitblit v1.9.1