This commit is contained in:
speech_asr 2023-03-15 17:22:01 +08:00
parent b4598f30a5
commit 7ee716759b
2 changed files with 1 additions and 4 deletions

View File

@ -2,7 +2,7 @@ from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
inference_diar_pipline = pipeline(
task=Tasks.auto_speech_recognition,
task=Tasks.speaker_diarization,
model='damo/speech_diarization_eend-ola-en-us-callhome-8k',
model_revision="v1.0.0",
)

View File

@ -234,9 +234,6 @@ def inference_launch(**kwargs):
elif mode == "rnnt":
from funasr.bin.asr_inference_rnnt import inference_modelscope
return inference_modelscope(**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