Persian T5 Formality Transfer

🚀 波斯语正式风格转换模型

这是一个用于波斯语的正式风格转换模型，可将口语化文本转换为正式文本。该模型基于波斯语单语T5模型和波斯语T5释义器构建。

⚠️ 重要提示

此模型仍在开发中，因此其输出效果可能不太理想。不过，你可以尝试为解码器设置不同的值以获得更好的结果。更多信息请查看此链接。

🚀 快速开始

📦 安装指南

>>> pip install transformers

💻 使用示例

基础用法

>>> from transformers import (T5ForConditionalGeneration, AutoTokenizer, pipeline)
>>> import torch

model_path = 'erfan226/persian-t5-formality-transfer'
model = T5ForConditionalGeneration.from_pretrained(model_path)
tokenizer = AutoTokenizer.from_pretrained(model_path)
pipe = pipeline(task='text2text-generation', model=model, tokenizer=tokenizer)

def paraphrase(text):
  for j in range(3):
    out = pipe(text, encoder_no_repeat_ngram_size=4, do_sample=True, num_beams=5, max_length=128)[0]['generated_text']
    print("Paraphrase:", out)

text = "من با دوستام میرم بازی"
print("Original:", text)
paraphrase(text)

# Original: من با دوستام میرم بازی
# Paraphrase: دوست دارم با دوستانم بازی کنم.
# Paraphrase: من با دوستانم میرم...
# Paraphrase: من با دوستام بازی می کنم.

📚 详细文档

该模型可将波斯语的口语化文本转换为正式文本。你可以通过调用paraphrase函数，传入需要转换的口语化文本，模型会输出不同的正式文本转换结果。在使用时，你可以根据需求调整解码器的参数，如encoder_no_repeat_ngram_size、do_sample、num_beams和max_length等，以获得更符合你期望的转换结果。

📄 许可证

文档未提及相关许可证信息。

训练数据

文档暂未提供训练数据相关内容。