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