Spaces:
Sleeping
Sleeping
File size: 156 Bytes
3788fdf |
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)
|