abisee/cnn_dailymail
Viewer β’ Updated β’ 936k β’ 169k β’ 344
How to use AbdullahAlnemr1/flan-t5-summarizer with Transformers:
# Use a pipeline as a high-level helper
# Warning: Pipeline type "summarization" is no longer supported in transformers v5.
# You must load the model directly (see below) or downgrade to v4.x with:
# 'pip install "transformers<5.0.0'
from transformers import pipeline
pipe = pipeline("summarization", model="AbdullahAlnemr1/flan-t5-summarizer") # Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("AbdullahAlnemr1/flan-t5-summarizer")
model = AutoModelForSeq2SeqLM.from_pretrained("AbdullahAlnemr1/flan-t5-summarizer")This model is a fine-tuned version of google/flan-t5-base on the CNN/DailyMail dataset using the Hugging Face Transformers library.
Abstractive Summarization: Given a news article, generate a concise summary.
The model was fine-tuned on 20,000 training samples and validated/tested on 2,000 samples. Evaluation was performed using ROUGE metrics:
| Metric | Score |
|---|---|
| ROUGE-1 | 25.33 |
| ROUGE-2 | 11.96 |
| ROUGE-L | 20.68 |
| ROUGE-Lsum | 23.81 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
model = T5ForConditionalGeneration.from_pretrained("AbdullahAlnemr1/flan-t5-summarizer")
tokenizer = T5Tokenizer.from_pretrained("AbdullahAlnemr1/flan-t5-summarizer")
input_text = "summarize: The US president met with the Senate to discuss..."
inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
summary_ids = model.generate(inputs["input_ids"], max_length=128, num_beams=4, early_stopping=True)
print(tokenizer.decode(summary_ids[0], skip_special_tokens=True))
Base model
google/flan-t5-base