This model was obtained by fine-tuning
facebook/bart-large-xsum on Samsum dataset.
from transformers import pipeline summarizer = pipeline("summarization", model="lidiya/bart-large-xsum-samsum") conversation = '''Hannah: Hey, do you have Betty's number? Amanda: Lemme check Amanda: Sorry, can't find it. Amanda: Ask Larry Amanda: He called her last time we were at the park together Hannah: I don't know him well Amanda: Don't be shy, he's very nice Hannah: If you say so.. Hannah: I'd rather you texted him Amanda: Just text him 🙂 Hannah: Urgh.. Alright Hannah: Bye Amanda: Bye bye ''' summarizer(conversation)
- Colab notebook: https://colab.research.google.com/drive/1dul0Sg-TTMy9xZCJzmDRajXbyzDwtYx6?usp=sharing
key value eval_rouge1 54.3921 eval_rouge2 29.8078 eval_rougeL 45.1543 eval_rougeLsum 49.942 test_rouge1 53.3059 test_rouge2 28.355 test_rougeL 44.0953 test_rougeLsum 48.9246
- Downloads last month