mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
26 lines
926 B
Python
Executable File
26 lines
926 B
Python
Executable File
#!/usr/bin/env python3
|
|
import argparse
|
|
|
|
from modelscope.pipelines import pipeline
|
|
from modelscope.utils.constant import Tasks
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description="download model configs",
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
|
|
)
|
|
parser.add_argument("--model_name",
|
|
type=str,
|
|
default="speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
|
|
help="model name in modelscope")
|
|
parser.add_argument("--model_revision",
|
|
type=str,
|
|
default="v1.0.3",
|
|
help="model revision in modelscope")
|
|
args = parser.parse_args()
|
|
|
|
inference_pipeline = pipeline(
|
|
task=Tasks.auto_speech_recognition,
|
|
model='damo/{}'.format(args.model_name),
|
|
model_revision=args.model_revision)
|