shravan

Running

shravan / topic.py

add summary topic sentiment features

ca9e7bd about 1 year ago

157 Bytes

	from transformers import pipeline

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