KJ24 commited on
Commit
073d270
·
verified ·
1 Parent(s): 59c38e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
 
 
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):