The model has been trained on a collection of 500k articles with headings. Its purpose is to create a one-line heading suitable for the given article.
Sample code with a WikiNews article:
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 = '''Very early yesterday morning, the United States President Donald Trump reported he and his wife First Lady Melania Trump tested positive for COVID-19. Officials said the Trumps' 14-year-old son Barron tested negative as did First Family and Senior Advisors Jared Kushner and Ivanka Trump.Trump took to social media, posting at 12:54 am local time (0454 UTC) on Twitter, "Tonight, [Melania] and I tested positive for COVID-19. We will begin our quarantine and recovery process immediately. We will get through this TOGETHER!" Yesterday afternoon Marine One landed on the White House's South Lawn flying Trump to Walter Reed National Military Medical Center (WRNMMC) in Bethesda, Maryland.Reports said both were showing "mild symptoms". Senior administration officials were tested as people were informed of the positive test. Senior advisor Hope Hicks had tested positive on Thursday.Presidential physician Sean Conley issued a statement saying Trump has been given zinc, vitamin D, Pepcid and a daily Aspirin. Conley also gave a single dose of the experimental polyclonal antibodies drug from Regeneron Pharmaceuticals.According to official statements, Trump, now operating from the WRNMMC, is to continue performing his duties as president during a 14-day quarantine. In the event of Trump becoming incapacitated, Vice President Mike Pence could take over the duties of president via the 25th Amendment of the US Constitution. The Pence family all tested negative as of yesterday and there were no changes regarding Pence's campaign events.'''
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)
Result:
Trump and First Lady Melania Test Positive for COVID-19
Runs of Michau t5-base-en-generate-headline on huggingface.co
637
Total runs
-38
24-hour runs
-34
3-day runs
34
7-day runs
28
30-day runs
More Information About t5-base-en-generate-headline huggingface.co Model
t5-base-en-generate-headline huggingface.co
t5-base-en-generate-headline huggingface.co is an AI model on huggingface.co that provides t5-base-en-generate-headline's model effect (), which can be used instantly with this Michau t5-base-en-generate-headline model. huggingface.co supports a free trial of the t5-base-en-generate-headline model, and also provides paid use of the t5-base-en-generate-headline. Support call t5-base-en-generate-headline model through api, including Node.js, Python, http.
t5-base-en-generate-headline huggingface.co is an online trial and call api platform, which integrates t5-base-en-generate-headline's modeling effects, including api services, and provides a free online trial of t5-base-en-generate-headline, you can try t5-base-en-generate-headline online for free by clicking the link below.
Michau t5-base-en-generate-headline online free url in huggingface.co:
t5-base-en-generate-headline is an open source model from GitHub that offers a free installation service, and any user can find t5-base-en-generate-headline on GitHub to install. At the same time, huggingface.co provides the effect of t5-base-en-generate-headline install, users can directly use t5-base-en-generate-headline installed effect in huggingface.co for debugging and trial. It also supports api for free installation.
t5-base-en-generate-headline install url in huggingface.co:
