File size: 156 Bytes
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)