shravan / topic.py
peb-peb's picture
typo fix
3c953de
raw
history blame
156 Bytes
from transformers import pipeline
def topic_gen(text):
topic = pipeline("Text2Text Generation", model="knkarthick/TOPIC-DIALOGSUM")
return topic(text)