Transformers How to use vasanth0475/pegasus-bbc-news-custom with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("vasanth0475/pegasus-bbc-news-custom")
model = AutoModelForSeq2SeqLM.from_pretrained("vasanth0475/pegasus-bbc-news-custom", device_map="auto") 
