shravan / topic.py
peb-peb's picture
fix topic
9603f5e
raw
history blame
156 Bytes
from transformers import pipeline
def topic_gen(text):
topic = pipeline("text2text-generation", model="knkarthick/TOPIC-DIALOGSUM")
return topic(text)