语言: 波斯语
许可证: MIT
任务标签: 文本摘要
推理支持: 不支持

TasnimSum-MT5 摘要模型

本模型专为生成输入文本的摘要而微调。基于预训练模型pn-summary-mt5-small，在TasnimNews数据集上进行了微调。

使用方法

from transformers import AutoModelForSeq2SeqLM, MT5Tokenizer

model = AutoModelForSeq2SeqLM.from_pretrained('PardisSzah/Persian_Summarizer_MT5')
tokenizer = MT5Tokenizer.from_pretrained("PardisSzah/Persian_Summarizer_MT5")

text_to_summarize = """
（此处为波斯语原文，内容描述伊朗议会代表提交政府透明度法案的新闻）
"""

inputs = tokenizer(text_to_summarize, truncation=True, max_length=256, padding='max_length', return_tensors="pt")

# 生成摘要
summary_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=64, early_stopping=True)

# 解码摘要
summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)

print(summary)

# 输出示例: 尼沙布尔议会代表宣布向议会主席团提交了关于政府机构和公共非政府组织三权透明的紧急提案