Views
No views yet
1from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained("yuvraj/summarizer-cnndm")
4AutoModelWithLMHead.from_pretrained("yuvraj/summarizer-cnndm")
5
6summarizer = pipeline('summarization', model=model, tokenizer=tokenizer)
7summarizer("<Text to be summarized>")
8
9## Limitations and bias
10Trained on a small dataset