shravan

Sleeping

shravan / topic.py

fix topic

9603f5e 11 months ago

No virus

156 Bytes

	from transformers import pipeline

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