From 5134dfea33636a301d5f46dd86e17531536e2934 Mon Sep 17 00:00:00 2001
From: haoneng.lhn <haoneng.lhn@alibaba-inc.com>
Date: 星期四, 10 八月 2023 20:14:51 +0800
Subject: [PATCH] add mossformer inference recipe
---
egs_modelscope/speech_separation/speech_separation_mossformer_8k_pytorch/demo.py | 15 +++++++++++++++
1 files changed, 15 insertions(+), 0 deletions(-)
diff --git a/egs_modelscope/speech_separation/speech_separation_mossformer_8k_pytorch/demo.py b/egs_modelscope/speech_separation/speech_separation_mossformer_8k_pytorch/demo.py
new file mode 100644
index 0000000..d3674a8
--- /dev/null
+++ b/egs_modelscope/speech_separation/speech_separation_mossformer_8k_pytorch/demo.py
@@ -0,0 +1,15 @@
+import numpy
+import soundfile as sf
+from modelscope.pipelines import pipeline
+from modelscope.utils.constant import Tasks
+
+input = 'https://modelscope.cn/api/v1/models/damo/speech_separation_mossformer_8k_pytorch/repo?Revision=master&FilePath=examples/mix_speech1.wav'
+separation = pipeline(
+ Tasks.funasr_speech_separation,
+ model='damo/speech_separation_mossformer_8k_pytorch',
+ output_dir='./',
+ model_revision='v1.0.1')
+result = separation(audio_in=input)
+for i, signal in enumerate(result):
+ save_file = f'output_spk_{i+1}.wav'
+ sf.write(save_file, signal, 8000)
--
Gitblit v1.9.1