Views
No views yet
1from transformers import T5ForConditionalGeneration, T5Tokenizer
2
3# Load model and tokenizer
4model = T5ForConditionalGeneration.from_pretrained("ambrosfitz/summarize-grammar")
5tokenizer = T5Tokenizer.from_pretrained("ambrosfitz/summarize-grammar")
6
7# Prepare input
8text = "Your text here..."
9input_text = f"summarize: {text}"
10
11# Generate summary
12inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
13outputs = model.generate(**inputs, max_length=150, num_beams=4, length_penalty=2.0)
14summary = tokenizer.decode(outputs[0], skip_special_tokens=True)@misc{grammar-t5-summarizer,
author = {repo_owner},
title = {Grammar-Enhanced T5 Summarizer},
year = {2024},
publisher = {Hugging Face},
journal = {Hugging Face Model Hub},
howpublished = {https://huggingface.co/ambrosfitz/summarize-grammar}
}