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