Shubham170793 commited on
Commit
12d1bb1
·
verified ·
1 Parent(s): acebfd4

Update src/embeddings.py

Browse files
Files changed (1) hide show
  1. src/embeddings.py +2 -2
src/embeddings.py CHANGED
@@ -23,10 +23,10 @@ print(f"✅ Using Hugging Face cache at {CACHE_DIR}")
23
  # ----------------------------
24
  try:
25
  _model = SentenceTransformer(
26
- "intfloat/e5-large-v2", # ✅ Trained for retrieval-augmented QA
27
  cache_folder=CACHE_DIR
28
  )
29
- print("✅ Loaded model: intfloat/e5-large-v2")
30
  except Exception as e:
31
  print(f"⚠️ Model load failed ({e}), falling back to MiniLM.")
32
  _model = SentenceTransformer(
 
23
  # ----------------------------
24
  try:
25
  _model = SentenceTransformer(
26
+ "intfloat/e5-small-v2", # ✅ Trained for retrieval-augmented QA
27
  cache_folder=CACHE_DIR
28
  )
29
+ print("✅ Loaded model: intfloat/e5-small-v2")
30
  except Exception as e:
31
  print(f"⚠️ Model load failed ({e}), falling back to MiniLM.")
32
  _model = SentenceTransformer(