模型描述

PEGASUS 对概述进行了微调

安装所需的"sentencepiece"库进行分词器函数

pip install sentencepiece

模型实际应用?

import torch
from transformers import PegasusForConditionalGeneration, PegasusTokenizer
model_name = 'tuner007/pegasus_summarizer'
torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
tokenizer = PegasusTokenizer.from_pretrained(model_name)
model = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)

def get_response(input_text):
  batch = tokenizer([input_text],truncation=True,padding='longest',max_length=1024, return_tensors="pt").to(torch_device)
  gen_out = model.generate(**batch,max_length=128,num_beams=5, num_return_sequences=1, temperature=1.5)
  output_text = tokenizer.batch_decode(gen_out, skip_special_tokens=True)
  return output_text

示例：

context = """"印度捕手兼击球手Rishabh Pant表示，在周三进行的对阵英格兰的第三场测试中，观众中的某人把球扔向投手Mohammed Siraj时，印度队队长Virat Kohli感到"不快乐"。Pant在当天比赛结束后的虚拟新闻发布会上透露，这一事件使印度队队长Virat Kohli感到"不快乐"。"我想，有人在内部朝Siraj扔了个球，所以他（Kohli）很不高兴，" Pant说。"你可以随便说自己想说的话，但是不要朝着场上的球员扔东西之类的。我想，这对于板球来说不好。"他补充道。

在第三场测试的开局阶段，似乎有一部分观众问了Siraj比赛的比分，以戏弄这位投手。然而，印度球员给出了一个出色的回答，他向观众比了1-0（印度领先测试系列赛）。本月早些时候，在第二场测试比赛中，观众的一些非法行为在秀中引发了不良反应，有人向印度击球手KL Rahul扔香槟塞子。Kohli也插手其中，他被看到向开局的击球手打手势，以了解更多关于这件事的情况。一局之后，电视画面显示许多香槟塞子被扔在球场内，印度球员显然感到沮丧。

回到比赛，印度78分全员出局后，开局罗里·伯恩斯和哈西卜·哈米德确保英格兰在进行中的第三场测试的开局之日占据了上风。结束时，英格兰的得分为120/0，东道主将领先42分。对于英格兰队，伯恩斯（52*）和哈米德（60*）目前在击球区不败。

谈到开局时的球场情况，Pant说：“他们使用了重型滚筒，球场更加安定，他们也打得漂亮，”他说。“但是当我们打球时，球场稍微软一些，他们打在好的地方，但我们本可以更好地应对。”

两名英格兰板球手设法挨过最后一个段落，主队以全部十个击球手保持了开局，将领先优势扩大到42分。（北美新闻社）"""

get_response(context)

输出：

Team India wicketkeeper-batsman Rishabh Pant表示，在进行中的第三场测试中，有人向投手Mohammed Siraj扔球时，队长Virat Kohli感到"不快乐"。Pant补充说："你可以随便说自己想说的话，但请不要朝着场上的球员扔东西之类的事情。我想，这对于板球来说不好。"

Inshort

印度捕手兼击球手Rishabh Pant透露，在对阵英格兰的利兹测试的第一天，队长Virat Kohli对观众感到不满，因为有人向投手Mohammed Siraj扔了一颗球。Pant补充说："你可以随便说自己想说的话，但请不要朝着场上的球员扔东西之类的事情。这对板球来说不好。"

Created by Arpit Rajauria