Radosław Wolnik commited on
Commit
6042737
·
1 Parent(s): b26d149

new methot of generation

Browse files
Files changed (1) hide show
  1. ChatAI/chat_ai.py +2 -2
ChatAI/chat_ai.py CHANGED
@@ -1,3 +1,3 @@
1
- from transformers import pipeline
2
 
3
- pipe = pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")
 
1
+ from huggingface_hub import InferenceClient
2
 
3
+ pipe = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3")