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 | 13 +++++++++----
1 files changed, 9 insertions(+), 4 deletions(-)
diff --git a/examples/industrial_data_pretraining/seaco_paraformer/demo.py b/examples/industrial_data_pretraining/seaco_paraformer/demo.py
index e9e226d..804acdd 100644
--- a/examples/industrial_data_pretraining/seaco_paraformer/demo.py
+++ b/examples/industrial_data_pretraining/seaco_paraformer/demo.py
@@ -11,20 +11,25 @@
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",
+ spk_model="damo/speech_campplus_sv_zh-cn_16k-common",
+ spk_model_revision="v2.0.2",
)
# 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)
@@ -33,8 +38,8 @@
# example3
import soundfile
-import os
+
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