Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,9 @@ os.environ["HF_HUB_CACHE"] = CACHE_DIR
|
|
25 |
|
26 |
# ✅ Configuration du modèle d’embedding local (ex: BGE / Nomic / GTE etc.)
|
27 |
MODEL_NAME = "BAAI/bge-small-en-v1.5"
|
28 |
-
|
|
|
|
|
29 |
model = AutoModel.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
|
30 |
|
31 |
def get_embedding(text: str):
|
|
|
25 |
|
26 |
# ✅ Configuration du modèle d’embedding local (ex: BGE / Nomic / GTE etc.)
|
27 |
MODEL_NAME = "BAAI/bge-small-en-v1.5"
|
28 |
+
|
29 |
+
# tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
|
30 |
+
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=os.getenv("HF_HOME"))
|
31 |
model = AutoModel.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
|
32 |
|
33 |
def get_embedding(text: str):
|