infer.py 389 B

123456789101112131415
  1. from modelscope.pipelines import pipeline
  2. from modelscope.utils.constant import Tasks
  3. import sys
  4. # Define extraction pipeline
  5. inference_sv_pipline = pipeline(
  6. task=Tasks.speaker_verification,
  7. model='damo/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch',
  8. output_dir=sys.argv[2],
  9. )
  10. # Extract speaker embedding
  11. rec_result = inference_sv_pipline(
  12. audio_in=sys.argv[1],
  13. )