Update app.py

app.py

@@ -5,14 +5,15 @@ import torch
 import torch.nn.functional as F
 import os
 
-# 📁 Définir le cache autorisé
+# 📁 Rediriger les caches HF vers un dossier autorisé
 CACHE_DIR = "/data"
 os.environ["HF_HOME"] = CACHE_DIR
 os.environ["TRANSFORMERS_CACHE"] = CACHE_DIR
+os.environ["HF_MODULES_CACHE"] = CACHE_DIR  # ✅ <--- ajoute cette ligne
 
 app = FastAPI()
 
-# ✅ Modèle avec custom code : activer trust_remote_code
+# ✅ Modèle avec custom code
 MODEL_NAME = "nomic-ai/nomic-embed-text-v1"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR, trust_remote_code=True)
 model = AutoModel.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR, trust_remote_code=True)