Downloads · 30 days
29
8% of all-time downloads
kubershahi/pegasus-inshorts
pegasus-inshorts is a machine learning model from kubershahi. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
Given a news article, generate a summary of two-to-three sentences and a headline for the article. The summary should be abstractive rather than extractive. In abstractive summarization, new sentences are generated as…
Downloads · 30 days
29
8% of all-time downloads
All-time downloads
373
Public
Repo size
4.6 GB
Likes
0
Public
Click a slice to open those files.
.bin2.3 GB · 100%
From the Hugging Face model README
Given a news article, generate a summary of two-to-three sentences and a headline for the article. The summary should be abstractive rather than extractive. In abstractive summarization, new sentences are generated as part of the summary and the sentences in the summary might not be present in the news article.
This model builds on the google/pegasus-large model by finetuning it on a custom summary-headline dataset called inshorts. After finetuning, to generate an appropriate headline of an article, get the summary of the article first from the pegasus-large model and then pass the summary through this model. The two-way approach was taken to get apt headline from summary rather then generating the headline from the pegasus-large itself.
For more details about the project, click here.