Upload database.py

app/database.py

@@ -1,4 +1,14 @@
 import os
+# Règle d’or : toute variable d’environnement qui influence le cache Hugging Face doit être
+# définie avant d’importer datasets ou transformers, sinon elle sera ignorée.
+cache_dir = "/tmp"
+os.makedirs(cache_dir, exist_ok=True)
+
+# Rediriger le cache HF globalement
+os.environ["HF_HOME"] = cache_dir
+os.environ["HF_DATASETS_CACHE"] = os.path.join(cache_dir, "datasets")
+os.environ["TRANSFORMERS_CACHE"] = os.path.join(cache_dir, "transformers")
+
 from typing import List, Dict, Any
 import duckdb
 import faiss
@@ -21,13 +31,6 @@ FAISS_INDEX_FILE = "faiss_index.bin"
 MODEL_NAME = "intfloat/multilingual-e5-small"
 CROSS_ENCODER_NAME = "cross-encoder/ms-marco-MiniLM-L12-v2"
 
-cache_dir = "/tmp"
-os.makedirs(cache_dir, exist_ok=True)
-# Rediriger le cache HF globalement
-os.environ["HF_HOME"] = cache_dir
-os.environ["HF_DATASETS_CACHE"] = cache_dir
-os.environ["TRANSFORMERS_CACHE"] = cache_dir
-
 # Téléchargement des fichiers Parquet depuis Hugging Face
 articles_parquet = hf_hub_download(
     repo_id=REPO_ID,