change name

This commit is contained in:
九耳 2023-03-30 17:29:12 +08:00
parent 5afbcdc395
commit 3cd71a385a
3 changed files with 5 additions and 5 deletions

View File

@ -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="跨境河流是养育沿岸人民的生命之源长期以来为帮助下游地区防灾减灾中方技术人员在上游地区极为恶劣的自然条件下克服巨大困难甚至冒着生命危险向印方提供汛期水文资料处理紧急事件中方重视印方在跨境河流问题上的关切愿意进一步完善双方联合工作机制凡是中方能做的我们都会去做而且会做得更好我请印度朋友们放心中国在上游的任何开发利用都会经过科学规划和论证兼顾上下游的利益"

View File

@ -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

View File

@ -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",