damo-csanmt-en-zh-base

The original model is at: https://www.modelscope.cn/models/damo/nlp_csanmt_translation_en2zh_base/summary

Usage

Option 1: use modelscope library

pip install modelscope subword_nmt sacremoses jieba

from huggingface_hub import snapshot_download
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks

model_dir = snapshot_download('modelscope-unofficial/damo-csanmt-en-zh-base')

pipe = pipeline(task=Tasks.translation, model=model_dir)

input_sequence = 'Elon Musk, co-founder and chief executive officer of Tesla Motors.'
outputs = pipe(input=input_sequence)

print(outputs['translation'])  # 'Elon Musk，特斯拉汽车公司联合创始人兼首席执行官。'


batch_input_sequences = [
    "Elon Musk, co-founder and chief executive officer of Tesla Motors.",
    "What's the weather like today?"
]
input_sequence = '<SENT_SPLIT>'.join(batch_input_sequences)
outputs = pipe(input=input_sequence)
print(outputs['translation'].split('<SENT_SPLIT>'))