from transformers import pipeline def topic_gen(text): topic = pipeline("Text2Text Generation", model="knkarthick/TOPIC-DIALOGSUM") return topic(text)