1.环境
Python 3.10.14
2.完整代码
from transformers import pipeline, AutoModelWithLMHead, AutoTokenizer
modelName = "./damo/opus-mt-zh-en" # 模型名
model = AutoModelWithLMHead.from_pretrained(modelName)
tokenizer = AutoTokenizer.from_pretrained(modelName)
translation = pipeline('translation_zh_to_en', model=model, tokenizer=tokenizer)
res = translation("你好,我是你的哥哥")[0]['translation_text']
print(res)
3.效果
4.问题
1.英文转中文就用opus-mt-en-ch
发布评论