Duplicated from peb-peb/shravan
ca9e7bd 9603f5e ca9e7bd
1
2
3
4
5
6
from transformers import pipeline def topic_gen(text): topic = pipeline("text2text-generation", model="knkarthick/TOPIC-DIALOGSUM") return topic(text)