Shubham170793 commited on
Commit
83d7123
·
verified ·
1 Parent(s): 3a765a5

Update src/qa.py

Browse files
Files changed (1) hide show
  1. src/qa.py +3 -3
src/qa.py CHANGED
@@ -29,11 +29,11 @@ os.environ.update({
29
  })
30
 
31
  # ==========================================================
32
- # 2️⃣ Embedding Model (E5-small-v2)
33
  # ==========================================================
34
  try:
35
- _query_model = SentenceTransformer("intfloat/e5-small-v2", cache_folder=CACHE_DIR)
36
- print("✅ Loaded embedding model: intfloat/e5-small-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-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)