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é
+# 📁 Rediriger le cache HF vers un dossier autorisé
 CACHE_DIR = "/data"
-os.environ['HF_HOME'] = CACHE_DIR
-os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
+os.environ["HF_HOME"] = CACHE_DIR
+os.environ["TRANSFORMERS_CACHE"] = CACHE_DIR
 
 app = FastAPI()
 
-# ✅ Charger modèle et tokenizer avec cache_dir précisé explicitement
-MODEL_NAME = "thenlper/gte-small"
+# 📌 Nouveau modèle Jina Embeddings v3
+MODEL_NAME = "jinaai/jina-embeddings-v3-base-en"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
 model = AutoModel.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
 
@@ -27,4 +27,4 @@ async def embed_text(payload: EmbedInput):
         outputs = model(**inputs)
         embeddings = outputs.last_hidden_state[:, 0]  # CLS token
         normalized = F.normalize(embeddings, p=2, dim=1)
-    return {'embedding': normalized[0].tolist()}
+    return {"embedding": normalized[0].tolist()}