Summarization : Conversation

I am new in this area. Please advise some good models to generate a summary on conversation between two persons. Thank you!

hey @MattJan a good place to start would be by looking at models fine-tuned on the samsum dataset (dialogues between two people + their summary): Hugging Face – The AI community building the future.

if you want to fine-tune your own model, a good start would be to use a pegasus model that has already be trained for summarisation, e.g. google/pegasus-cnn_dailymail · Hugging Face

1 Like