Views
No views yet
bwlw3127/clickbait-bart-dailymailfacebook/bart-base to generate clickbait-style headlines from DailyMail articles.facebook/bart-base and fine-tuned on scraped DailyMail article/headline pairs.facebook/bart-base1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3model_id = "bwlw3127/clickbait-bart-dailymail"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
6
7article = "Scientists have discovered a new exoplanet twice the size of Earth..."
8inputs = tokenizer("generate clickbait headline: " + article,
9 return_tensors="pt", truncation=True, max_length=512)
10
11outputs = model.generate(**inputs, max_length=24, num_beams=4, early_stopping=True)
12print(tokenizer.decode(outputs[0], skip_special_tokens=True))