Radosław Wolnik commited on
Commit
9a675b3
·
1 Parent(s): a54eb10
Files changed (2) hide show
  1. ChatAI/chat_ai.py +1 -1
  2. requirements.txt +1 -1
ChatAI/chat_ai.py CHANGED
@@ -1,4 +1,4 @@
1
  from transformers import pipeline
2
 
3
 
4
- pipe = pipeline("text2text-generation", model="google/flan-t5-xxl")
 
1
  from transformers import pipeline
2
 
3
 
4
+ pipe = pipeline("text-generation", model="microsoft/phi-2")
requirements.txt CHANGED
@@ -2,4 +2,4 @@ transformers
2
  huggingface_hub==0.25.2
3
  discord.py
4
  discord
5
- TensorFlow>=2.0
 
2
  huggingface_hub==0.25.2
3
  discord.py
4
  discord
5
+ torch