| | |
| | | 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", |
| | | ) |
| | |
| | | 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 |