Update app.py

app.py

@@ -8,6 +8,8 @@ from litellm import completion
 from langchain_community.embeddings import HuggingFaceEmbeddings
 
 from huggingface_hub import login
+import pickle
+
 
 
 token=os.getenv("HF_TOKEN")
@@ -49,6 +51,19 @@ index = faiss.read_index(doc_path)
 if not index:
     print("Problème index")
 
+# Télécharger le fichier depuis le Hub
+doc_path = hf_hub_download(repo_id="xavierbarbier/rag_ngap", filename="resource/fichier_chunks.pkl", repo_type="space")
+
+# Charger les chunks depuis le fichier pickle
+with open(doc_path, "rb") as f:
+    chunks = pickle.load(f)
+
+if not chunks:
+    print("Problème chunks")
+
+if len(chunks):
+    print("Problème len chunks")
+
 def qa(question):
     """Fonction principale pour répondre à la question en utilisant RAG avec jusqu'à 5 chunks."""