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

Update src/qa.py

Browse files
Files changed (1) hide show
  1. src/qa.py +6 -3
src/qa.py CHANGED
@@ -29,11 +29,14 @@ os.environ.update({
29
  })
30
 
31
  # ==========================================================
32
- # 2️⃣ Embedding Model (E5-large-v2)
33
  # ==========================================================
34
  try:
35
- _query_model = SentenceTransformer("intfloat/e5-large-v2", cache_folder=CACHE_DIR)
36
- print("✅ Loaded embedding model: intfloat/e5-large-v2")
 
 
 
37
  except Exception as e:
38
  print(f"⚠️ Embedding load failed ({e}), using MiniLM fallback")
39
  _query_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder=CACHE_DIR)
 
29
  })
30
 
31
  # ==========================================================
32
+ # 2️⃣ Embedding Model (E5-small-v2)
33
  # ==========================================================
34
  try:
35
+ _query_model = SentenceTransformer(
36
+ "intfloat/e5-small-v2", # ⚡ Faster, 384-dim embeddings
37
+ cache_folder=CACHE_DIR
38
+ )
39
+ print("✅ Loaded embedding model: intfloat/e5-small-v2 (fast mode)")
40
  except Exception as e:
41
  print(f"⚠️ Embedding load failed ({e}), using MiniLM fallback")
42
  _query_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder=CACHE_DIR)