关于该模型

该模型基于50万篇带标题的文章训练而成，其核心功能是为给定文章生成合适的单行标题。

以下是一段使用维基新闻文章的示例代码：

import torch
from transformers import T5ForConditionalGeneration,T5Tokenizer

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

model = T5ForConditionalGeneration.from_pretrained("Michau/t5-base-en-generate-headline")
tokenizer = T5Tokenizer.from_pretrained("Michau/t5-base-en-generate-headline")
model = model.to(device)

article = '''
昨日凌晨，美国总统唐纳德·特朗普宣布他与第一夫人梅拉尼娅·特朗普的新冠病毒检测结果呈阳性。官方表示，特朗普14岁的儿子巴伦以及第一家庭成员兼高级顾问贾里德·库什纳、伊万卡·特朗普检测结果为阴性。
特朗普于当地时间凌晨0点54分（UTC时间4点54分）在推特发文称："今晚，[梅拉尼娅]与我新冠病毒检测呈阳性。我们将立即开始隔离与康复过程。我们将共同度过难关！"昨日下午，海军陆战队一号直升机降落在白宫南草坪，将特朗普送往马里兰州贝塞斯达的沃尔特·里德国家军事医疗中心。
据报道，两人均出现"轻微症状"。在阳性检测结果公布后，多名政府高级官员接受了检测。高级顾问霍普·希克斯已于周四检测呈阳性。
总统医生肖恩·康利发表声明称，特朗普已接受锌剂、维生素D、法莫替丁和每日阿司匹林治疗，并单次注射了再生元制药的实验性多克隆抗体药物。
根据官方声明，特朗普将在医疗中心继续履行总统职责，进行14天隔离。若特朗普丧失履职能力，副总统迈克·彭斯可依据美国宪法第25修正案接任总统职责。截至昨日，彭斯全家检测呈阴性，其竞选行程未受影响。
'''

text =  "headline: " + article

max_len = 256

encoding = tokenizer.encode_plus(text, return_tensors = "pt")
input_ids = encoding["input_ids"].to(device)
attention_masks = encoding["attention_mask"].to(device)

beam_outputs = model.generate(
    input_ids = input_ids,
    attention_mask = attention_masks,
    max_length = 64,
    num_beams = 3,
    early_stopping = True,
)

result = tokenizer.decode(beam_outputs[0])
print(result)

运行结果：

特朗普与第一夫人梅拉尼娅新冠病毒检测呈阳性