Programmes commited on
Commit
4c6516b
·
verified ·
1 Parent(s): 371d3bd

Update rag_utils.py

Browse files
Files changed (1) hide show
  1. rag_utils.py +1 -0
rag_utils.py CHANGED
@@ -6,6 +6,7 @@ import re
6
  from sentence_transformers import SentenceTransformer
7
  from transformers import AutoTokenizer # Ajouté pour la gestion des tokens
8
  from huggingface_hub import InferenceClient
 
9
 
10
  # Chargement du modèle
11
  client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1", token=os.environ.get("edup2"))
 
6
  from sentence_transformers import SentenceTransformer
7
  from transformers import AutoTokenizer # Ajouté pour la gestion des tokens
8
  from huggingface_hub import InferenceClient
9
+ import os
10
 
11
  # Chargement du modèle
12
  client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1", token=os.environ.get("edup2"))