From 111450d5e67ed01e3e7de088fe5bb3115dc7eeff Mon Sep 17 00:00:00 2001
From: mengzhe.cmz <mengzhe.cmz@alibaba-inc.com>
Date: 星期四, 23 三月 2023 19:55:15 +0800
Subject: [PATCH] update infer.py
---
egs_modelscope/vad/speech_fsmn_vad_zh-cn-16k-common/infer_online.py | 10 +++++++---
1 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/egs_modelscope/vad/speech_fsmn_vad_zh-cn-16k-common/infer_online.py b/egs_modelscope/vad/speech_fsmn_vad_zh-cn-16k-common/infer_online.py
index bcf764b..d70ed25 100644
--- a/egs_modelscope/vad/speech_fsmn_vad_zh-cn-16k-common/infer_online.py
+++ b/egs_modelscope/vad/speech_fsmn_vad_zh-cn-16k-common/infer_online.py
@@ -1,16 +1,20 @@
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
+from modelscope.utils.logger import get_logger
+import logging
+logger = get_logger(log_level=logging.CRITICAL)
+logger.setLevel(logging.CRITICAL)
import soundfile
-
if __name__ == '__main__':
output_dir = None
inference_pipline = pipeline(
task=Tasks.voice_activity_detection,
model="damo/speech_fsmn_vad_zh-cn-16k-common-pytorch",
- model_revision='v1.1.9',
- output_dir=None,
+ model_revision='v1.2.0',
+ output_dir=output_dir,
batch_size=1,
+ mode='online',
)
speech, sample_rate = soundfile.read("./vad_example_16k.wav")
speech_length = speech.shape[0]
--
Gitblit v1.9.1