Shubham170793 commited on
Commit
6df27c5
Β·
verified Β·
1 Parent(s): e00eaea

Update src/embeddings.py

Browse files
Files changed (1) hide show
  1. src/embeddings.py +1 -1
src/embeddings.py CHANGED
@@ -23,7 +23,7 @@ print(f"βœ… Using Hugging Face cache at {CACHE_DIR}")
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")
 
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-small-v2")