From 46d9cc0b374470ca03339d63a38d213eb4fd889e Mon Sep 17 00:00:00 2001 From: speech_asr Date: Tue, 14 Mar 2023 22:54:24 +0800 Subject: [PATCH] update --- .../infer.py | 8 ++++++++ funasr/bin/diar_inference_launch.py | 3 +++ 2 files changed, 11 insertions(+) create mode 100644 egs_modelscope/speaker_diarization/speech_diarization_eend-ola-en-us-callhome-8k/infer.py diff --git a/egs_modelscope/speaker_diarization/speech_diarization_eend-ola-en-us-callhome-8k/infer.py b/egs_modelscope/speaker_diarization/speech_diarization_eend-ola-en-us-callhome-8k/infer.py new file mode 100644 index 000000000..fa4e8bf04 --- /dev/null +++ b/egs_modelscope/speaker_diarization/speech_diarization_eend-ola-en-us-callhome-8k/infer.py @@ -0,0 +1,8 @@ +from modelscope.pipelines import pipeline +from modelscope.utils.constant import Tasks + +inference_diar_pipline = pipeline( + task=Tasks.speaker_diarization, + model='damo/speech_diarization_eend-ola-en-us-callhome-8k', +) +results = inference_diar_pipline(audio_in="https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_data/record.wav") \ No newline at end of file diff --git a/funasr/bin/diar_inference_launch.py b/funasr/bin/diar_inference_launch.py index 7738f4f4f..70bb947b4 100755 --- a/funasr/bin/diar_inference_launch.py +++ b/funasr/bin/diar_inference_launch.py @@ -142,6 +142,9 @@ def inference_launch(mode, **kwargs): else: kwargs["param_dict"] = param_dict return inference_modelscope(mode=mode, **kwargs) + elif mode == "eend-ola": + from funasr.bin.eend_ola_inference import inference_modelscope + return inference_modelscope(mode=mode, **kwargs) else: logging.info("Unknown decoding mode: {}".format(mode)) return None