Programmes commited on
Commit
48391f9
·
verified ·
1 Parent(s): 4c6516b

Update rag_utils.py

Browse files
Files changed (1) hide show
  1. rag_utils.py +2 -2
rag_utils.py CHANGED
@@ -11,8 +11,8 @@ import os
11
  # Chargement du modèle
12
  client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1", token=os.environ.get("edup2"))
13
 
14
- # Chargement du tokenizer (même base que Falcon)
15
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
16
 
17
  def load_faiss_index(index_path="faiss_index/faiss_index.faiss", doc_path="faiss_index/documents.pkl"):
18
  index = faiss.read_index(index_path)
 
11
  # Chargement du modèle
12
  client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1", token=os.environ.get("edup2"))
13
 
14
+ # Chargement du tokenizer
15
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1", token=os.environ.get("edup2"))
16
 
17
  def load_faiss_index(index_path="faiss_index/faiss_index.faiss", doc_path="faiss_index/documents.pkl"):
18
  index = faiss.read_index(index_path)