使用模型Helsinki-NLP/opus-mt-en-zh实现英译中

原创

已于 2024-02-13 13:53:08 修改 · 3.2k 阅读

标签

#自然语言处理 #机器翻译 #人工智能

收录于

于 2024-02-13 13:21:28 首次发布

本文介绍了如何在GoogleColab中使用HuggingFace库（如AutoModel和MarianMTModel）加载预训练模型Helsinki-NLP/opus-mt-en-zh进行文本翻译，以及如何通过pipeline进行简单的翻译操作。

google codlab 运行需要5秒

from transformers import AutoModel , AutoTokenizer,MarianMTModel
from huggingface_hub.hf_api import HfFolder

HfFolder.save_token('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')

text ="Memphis is a series of models which advance human-data models, offering good performance without relying on proprietary model outputs"
mname ='Helsinki-NLP/opus-mt-en-zh'
tokenizer = AutoTokenizer.from_pretrained(mname)
model = MarianMTModel.from_pretrained