topic.py

from transformers import pipeline

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