Update app.py

app.py

@@ -5,15 +5,17 @@ import torch
 import torch.nn.functional as F
 import os
 
-# 💡 Correction ici
-os.environ['HF_HOME'] = '/data'
+# 👉 Rediriger le cache HF vers un dossier autorisé
+CACHE_DIR = "/data"
+os.environ['HF_HOME'] = CACHE_DIR
+os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
 
 app = FastAPI()
 
-# Charger le modèle
+# ✅ Charger modèle et tokenizer avec cache_dir précisé explicitement
 MODEL_NAME = "thenlper/gte-small"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModel.from_pretrained(MODEL_NAME)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
+model = AutoModel.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
 
 class EmbedInput(BaseModel):
     text: str