语言:

• 越南语 库名称: transformers 管道标签: 摘要生成

from transformers import AutoTokenizer, T5ForConditionalGeneration


device = 'cpu'

model_path = "CreatorPhan/ViSummary"
model = T5ForConditionalGeneration.from_pretrained(model_path).to(device)
tokenizer = AutoTokenizer.from_pretrained(model_path)


context = """
梅西重返巴萨交易有望成功的关键因素在于西甲联盟近期批准了巴萨提交的财务平衡计划。这帮助加泰罗尼亚俱乐部解决了最大难题。2021年夏天，由于俱乐部无法满足西甲财政规定，效力21年的梅西被迫离开巴萨。

梅西在与巴黎圣日耳曼两年合同到期后成为自由球员。他收到了多家俱乐部的邀约。据《The Athletic》报道，巴萨、国际迈阿密（美国）和一家沙特俱乐部正在争夺他。目前只有沙特方面提供了正式报价，年薪高达4亿美元。

尽管已35岁，梅西仍渴望回归造就他辉煌的巴萨。如今诺坎普由哈维执教——这位曾与梅西并肩作战并在其职业生涯初期给予指导的师兄正带领球队。
"""

tokens = tokenizer(f"为以下文本生成摘要: {context}", return_tensors='pt').input_ids
output = model.generate(tokens.to(device), max_new_tokens=170)[0]
predict = tokenizer.decode(output, skip_special_tokens=True)
print(len(predict.split()))
print(predict)