shravan

Sleeping

shravan / topic.py

fix topic

9603f5e over 1 year ago

156 Bytes

	from transformers import pipeline

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