typo fix
Browse files
topic.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
from transformers import pipeline
|
2 |
|
3 |
def topic_gen(text):
|
4 |
-
topic = pipeline("
|
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)
|