From 851e3e3ef83d0769d9bde172d8841f6b20e3e377 Mon Sep 17 00:00:00 2001
From: gaochangfeng <54253717+gaochangfeng@users.noreply.github.com>
Date: 星期三, 10 四月 2024 14:37:35 +0800
Subject: [PATCH] Gcf (#1605)

---
 examples/industrial_data_pretraining/sense_voice/demo.py |    8 ++++++--
 1 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/examples/industrial_data_pretraining/sense_voice/demo.py b/examples/industrial_data_pretraining/sense_voice/demo.py
index f6ad804..506b069 100644
--- a/examples/industrial_data_pretraining/sense_voice/demo.py
+++ b/examples/industrial_data_pretraining/sense_voice/demo.py
@@ -6,7 +6,11 @@
 from funasr import AutoModel
 
 model = AutoModel(model="/Users/zhifu/Downloads/modelscope_models/SenseVoice",
+                  vad_model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch",
+				  vad_kwargs={"max_single_segment_time": 30000},
                   )
-
-res = model.generate(input="https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav")
+task = "ASR"
+language = None
+input_wav = "https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav"
+res = model.generate(task=task, language=language, input=input_wav, batch_size_s=0,)
 print(res)

--
Gitblit v1.9.1