Radosław Wolnik commited on
Commit
b26d149
·
1 Parent(s): adc19e5
Files changed (1) hide show
  1. ChatAI/chat_ai.py +1 -2
ChatAI/chat_ai.py CHANGED
@@ -1,4 +1,3 @@
1
  from transformers import pipeline
2
 
3
-
4
- pipe = pipeline( model="facebook/blenderbot-400M-distill")
 
1
  from transformers import pipeline
2
 
3
+ pipe = pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")