Votre Nom commited on
Commit
2abc1bb
·
1 Parent(s): b898e35

fixing path data

Browse files
Files changed (1) hide show
  1. src/RAG.py +4 -4
src/RAG.py CHANGED
@@ -23,16 +23,16 @@ TOP_K = 10
23
  MAX_TOKENS_CONTEXT = 4000
24
  SYSTEM_PROMPT = (
25
  "Tu es un assistant expert en droit fiscal. "
26
- "Fais d’abord appel aux passages fournis pour répondre. "
27
  "Si ces passages sont insuffisants, utilise tes connaissances générales en le précisant clairement."
28
  )
29
 
30
  # —— CHARGEMENT DE L'INDEX ——
31
 
32
  # Télécharger les fichiers depuis le repo Hugging Face
33
- index_path = hf_hub_download(repo_id="Jordanche/fiscarag", filename=FAISS_INDEX_FILE)
34
- ids_path = hf_hub_download(repo_id="Jordanche/fiscarag", filename=IDS_PKL)
35
- chunks_path = hf_hub_download(repo_id="Jordanche/fiscarag", filename=CHUNKS_PKL)
36
 
37
  # Charger les fichiers
38
  index = faiss.read_index(index_path)
 
23
  MAX_TOKENS_CONTEXT = 4000
24
  SYSTEM_PROMPT = (
25
  "Tu es un assistant expert en droit fiscal. "
26
+ "Fais d'abord appel aux passages fournis pour répondre. "
27
  "Si ces passages sont insuffisants, utilise tes connaissances générales en le précisant clairement."
28
  )
29
 
30
  # —— CHARGEMENT DE L'INDEX ——
31
 
32
  # Télécharger les fichiers depuis le repo Hugging Face
33
+ index_path = hf_hub_download(repo_id="Jordanche/fiscarag", filename=FAISS_INDEX_FILE, repo_type="dataset")
34
+ ids_path = hf_hub_download(repo_id="Jordanche/fiscarag", filename=IDS_PKL, repo_type="dataset")
35
+ chunks_path = hf_hub_download(repo_id="Jordanche/fiscarag", filename=CHUNKS_PKL, repo_type="dataset")
36
 
37
  # Charger les fichiers
38
  index = faiss.read_index(index_path)