瀏覽代碼

Update infer_online.py

lingyunfly 3 年之前
父節點
當前提交
732d5ba514
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      egs_modelscope/vad/speech_fsmn_vad_zh-cn-16k-common/infer_online.py

+ 2 - 1
egs_modelscope/vad/speech_fsmn_vad_zh-cn-16k-common/infer_online.py

@@ -8,9 +8,10 @@ if __name__ == '__main__':
     inference_pipline = pipeline(
     inference_pipline = pipeline(
         task=Tasks.voice_activity_detection,
         task=Tasks.voice_activity_detection,
         model="damo/speech_fsmn_vad_zh-cn-16k-common-pytorch",
         model="damo/speech_fsmn_vad_zh-cn-16k-common-pytorch",
-        model_revision='v1.1.9',
+        model_revision='v1.2.0',
         output_dir=None,
         output_dir=None,
         batch_size=1,
         batch_size=1,
+        mode='online',
     )
     )
     speech, sample_rate = soundfile.read("./vad_example_16k.wav")
     speech, sample_rate = soundfile.read("./vad_example_16k.wav")
     speech_length = speech.shape[0]
     speech_length = speech.shape[0]