[Export] support gpu inference

wanchen.swc

2023-03-30 cfc4d402093060fe087424b0a6be4e2b2546eae8

1
2
3
4
5
6
7
8
9
10
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',
    model_revision="v1.0.0",
)
results = inference_diar_pipline(audio_in=["https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_data/record2.wav"])
print(results)