FunASR/funasr/runtime/python/onnxruntime/demo_punc_offline.py
2023-03-30 14:11:02 +08:00

10 lines
307 B
Python

from funasr_onnx import TargetDelayTransformer
model_dir = "/disk1/mengzhe.cmz/workspace/FunASR/funasr/export/damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch"
model = TargetDelayTransformer(model_dir)
text_in = "我们都是木头人不会讲话不会动"
result = model(text_in)
print(result)