🚀 基于Transformer的法英文本翻译
本项目借助Transformers库中的 ybanas/autotrain-fr-en-translate-51410121895
模型,实现将法语文本翻译成英语的功能。
🚀 快速开始
安装依赖
首先,你需要安装Transformers库,可通过以下命令完成:
pip install transformers
运行代码
- 将以下代码复制到一个
.py
文件中,例如 translation.py
:
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("ybanas/autotrain-fr-en-translate-51410121895")
model = AutoModelForSeq2SeqLM.from_pretrained("ybanas/autotrain-fr-en-translate-51410121895")
def translate_text(french_text: str) -> str:
"""
使用 ybanas/autotrain-fr-en-translate-51410121895 模型将法语文本翻译成英语。
参数:
french_text (str): 待翻译的法语文本。
返回:
str: 翻译后的英语文本。
"""
inputs = tokenizer(french_text, return_tensors="pt", padding=True, truncation=True)
outputs = model.generate(**inputs)
english_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
return english_text
if __name__ == "__main__":
french_text = "Les enfants aiment profiter des beaux jours"
english_text = translate_text(french_text)
print("French text:", french_text)
print("Translated English text:", english_text)
- 将
french_text
变量的值替换为你想要翻译的法语文本。
- 运行脚本:
python translation.py
翻译后的英语文本将显示在屏幕上。
💻 使用示例
基础用法
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("ybanas/autotrain-fr-en-translate-51410121895")
model = AutoModelForSeq2SeqLM.from_pretrained("ybanas/autotrain-fr-en-translate-51410121895")
def translate_text(french_text: str) -> str:
"""
使用 ybanas/autotrain-fr-en-translate-51410121895 模型将法语文本翻译成英语。
参数:
french_text (str): 待翻译的法语文本。
返回:
str: 翻译后的英语文本。
"""
inputs = tokenizer(french_text, return_tensors="pt", padding=True, truncation=True)
outputs = model.generate(**inputs)
english_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
return english_text
if __name__ == "__main__":
french_text = "Les enfants aiment profiter des beaux jours"
english_text = translate_text(french_text)
print("French text:", french_text)
print("Translated English text:", english_text)
📚 详细文档
本脚本使用 ybanas/autotrain-fr-en-translate-51410121895
模型将法语文本翻译成英语。该模型通过Transformers库中的 AutoTokenizer
和 AutoModelForSeq2SeqLM
类进行加载。translate_text
函数接收一个法语文本作为输入,并返回其英语翻译。
🔧 技术细节
模型训练信息
- 问题类型:翻译
- 模型ID:51410121895
- 二氧化碳排放量(克):86.9058
验证指标
指标 |
数值 |
损失值 |
1.455 |
SacreBLEU |
15.999 |
生成长度 |
15.299 |
数据集
本模型使用了 ybanas/autotrain-data-fr-en-translate
数据集进行训练。
支持语言