Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained("summarization")
4model = AutoModelForSeq2SeqLM.from_pretrained("summarization")
5
6input_text = "The movie was fantastic with a gripping storyline!"
7inputs = tokenizer.encode(input_text, return_tensors="pt")
8outputs = model(inputs)