Spaces:
Runtime error
Runtime error
Radosław Wolnik
commited on
Commit
·
f3d11f1
1
Parent(s):
8a3cd96
idk
Browse files- ChatAI/chat_ai.py +1 -2
- app.py +1 -0
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")
|
|
|
app.py
CHANGED
|
@@ -34,6 +34,7 @@ async def on_message(message):
|
|
| 34 |
|
| 35 |
response = ai("\n".join(messages), max_length=1900, truncation=True, pad_token_id=50256)
|
| 36 |
|
|
|
|
| 37 |
await channel.send(response[0]['generated_text'])
|
| 38 |
|
| 39 |
|
|
|
|
| 34 |
|
| 35 |
response = ai("\n".join(messages), max_length=1900, truncation=True, pad_token_id=50256)
|
| 36 |
|
| 37 |
+
|
| 38 |
await channel.send(response[0]['generated_text'])
|
| 39 |
|
| 40 |
|