peb-peb commited on
Commit
3c953de
1 Parent(s): ca9e7bd
Files changed (1) hide show
  1. topic.py +1 -1
topic.py CHANGED
@@ -1,5 +1,5 @@
1
  from transformers import pipeline
2
 
3
  def topic_gen(text):
4
- topic = pipeline(" Text2Text Generation", model="knkarthick/TOPIC-DIALOGSUM")
5
  return topic(text)
 
1
  from transformers import pipeline
2
 
3
  def topic_gen(text):
4
+ topic = pipeline("Text2Text Generation", model="knkarthick/TOPIC-DIALOGSUM")
5
  return topic(text)