shravan_duplicate

Sleeping

shravan_duplicate / topic.py

Duplicate from peb-peb/shravan

3788fdf almost 2 years ago

156 Bytes

	from transformers import pipeline

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