Jaita commited on
Commit
d0b39ff
·
verified ·
1 Parent(s): 23a78d8

Update kb_embed.py

Browse files
Files changed (1) hide show
  1. kb_embed.py +1 -1
kb_embed.py CHANGED
@@ -20,7 +20,7 @@ collection = client.get_or_create_collection(name="knowledge_base")
20
 
21
  # --- Embedding model ---
22
  # If you keep the model folder locally in the repo:
23
- model = SentenceTransformer(str(MODEL_DIR))
24
  # Or, if you prefer auto-download & cache (no local folder required):
25
  # model = SentenceTransformer("all-MiniLM-L6-v2")
26
 
 
20
 
21
  # --- Embedding model ---
22
  # If you keep the model folder locally in the repo:
23
+ model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder=str(MODEL_DIR))
24
  # Or, if you prefer auto-download & cache (no local folder required):
25
  # model = SentenceTransformer("all-MiniLM-L6-v2")
26