模型:
knkarthick/bart-large-xsum-samsum
这个模型是在 Samsum 个数据集上对facebook/bart-large-xsum进行微调得到的。
from transformers import pipeline summarizer = pipeline("summarization", model="knkarthick/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)