Hugging Face's logo

Spaces:
Rahul04052000
/
shravan
Sleeping

App Files Community
peb-peb commited on
Commit
9603f5e
•
1 Parent(s): e3d3d89

fix topic

Browse files
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)