九耳 3 anos atrás
pai
commit
3cd71a385a

+ 2 - 2
funasr/runtime/python/onnxruntime/demo_punc_offline.py

@@ -1,7 +1,7 @@
-from funasr_onnx import CT-Transformer
+from funasr_onnx import CT_Transformer
 
 model_dir = "/disk1/mengzhe.cmz/workspace/FunASR/funasr/export/damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch"
-model = CT-Transformer(model_dir)
+model = CT_Transformer(model_dir)
 
 text_in = "我们都是木头人不会讲话不会动"
 text_in="跨境河流是养育沿岸人民的生命之源长期以来为帮助下游地区防灾减灾中方技术人员在上游地区极为恶劣的自然条件下克服巨大困难甚至冒着生命危险向印方提供汛期水文资料处理紧急事件中方重视印方在跨境河流问题上的关切愿意进一步完善双方联合工作机制凡是中方能做的我们都会去做而且会做得更好我请印度朋友们放心中国在上游的任何开发利用都会经过科学规划和论证兼顾上下游的利益"

+ 2 - 2
funasr/runtime/python/onnxruntime/funasr_onnx/__init__.py

@@ -1,5 +1,5 @@
 # -*- encoding: utf-8 -*-
 from .paraformer_bin import Paraformer
 from .vad_bin import Fsmn_vad
-from .punc_bin import CT-Transformer
-#from .punc_bin import VadRealtimeCT-Transformer
+from .punc_bin import CT_Transformer
+#from .punc_bin import VadRealtimeCT_Transformer

+ 1 - 1
funasr/runtime/python/onnxruntime/funasr_onnx/punc_bin.py

@@ -12,7 +12,7 @@ from .utils.utils import (TokenIDConverter, split_to_mini_sentence,code_mix_spli
 logging = get_logger()
 
 
-class CT-Transformer():
+class CT_Transformer():
     def __init__(self, model_dir: Union[str, Path] = None,
                  batch_size: int = 1,
                  device_id: Union[str, int] = "-1",