| | |
|---|
| | | |
|---|
| | | parser.add_argument("--punc_model", |
|---|
| | | type=str, |
|---|
| | | default="damo/punc_ct-transformer_zh-cn-common-vad_realtime-vocab272727", |
|---|
| | | default="", |
|---|
| | | help="model from modelscope") |
|---|
| | | parser.add_argument("--ngpu", |
|---|
| | | type=int, |
|---|
| | |
|---|
| | | param_dict=param_dict_asr, |
|---|
| | | ngpu=args.ngpu, |
|---|
| | | ) |
|---|
| | | if args.punc_model is not None: |
|---|
| | | if args.punc_model != "": |
|---|
| | | param_dict_punc = {'cache': list()} |
|---|
| | | inference_pipeline_punc = pipeline( |
|---|
| | | task=Tasks.punctuation, |
|---|
| | |
|---|
| | | rec_result = inference_pipeline_asr(audio_in=audio_in) |
|---|
| | | if inference_pipeline_punc is not None and 'text' in rec_result: |
|---|
| | | rec_result = inference_pipeline_punc(text_in=rec_result['text'], param_dict=param_dict_punc) |
|---|
| | | print(rec_result["text"]) |
|---|
| | | print(rec_result["text"] if "text" in rec_result else rec_result) |
|---|
| | | time.sleep(0.1) |
|---|
| | | time.sleep(0.1) |
|---|
| | | |
|---|
