Votre Nom commited on
Commit ·
2abc1bb
1
Parent(s): b898e35
fixing path data
Browse files- 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
|
| 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)
|