Update rag_utils.py

rag_utils.py

@@ -7,11 +7,11 @@ from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer  # Ajouté pour la gestion des tokens
 from huggingface_hub import InferenceClient
 
-# Chargement du modèle Falcon-7B
-client = InferenceClient("tiiuae/falcon-7b-instruct")
+# Chargement du modèle 
+client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1", token=os.environ.get("edup2"))
 
 # Chargement du tokenizer (même base que Falcon)
-tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-7b-instruct")
+tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
 
 def load_faiss_index(index_path="faiss_index/faiss_index.faiss", doc_path="faiss_index/documents.pkl"):
     index = faiss.read_index(index_path)
@@ -73,5 +73,5 @@ Réponse :"""
     print("===== RÉPONSE REÇUE =====")
     print(response)
 
-    return response.get("generated_text", response)  # selon format du retour
+    return response  # selon format du retour