Update app.py

app.py

@@ -5,15 +5,15 @@ import torch
 import torch.nn.functional as F
 import os
 
-# 📁 Rediriger le cache HF vers un dossier autorisé
+# 📁 Définir le dossier de cache autorisé
 CACHE_DIR = "/data"
 os.environ["HF_HOME"] = CACHE_DIR
 os.environ["TRANSFORMERS_CACHE"] = CACHE_DIR
 
 app = FastAPI()
 
-# 📌 Nouveau modèle Jina Embeddings v3
-MODEL_NAME = "jinaai/jina-embeddings-v3-base-en"
+# ✅ Charger le modèle nomic-embed-text-v1
+MODEL_NAME = "nomic-ai/nomic-embed-text-v1"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
 model = AutoModel.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)