Update src/embeddings.py
Browse files- src/embeddings.py +2 -2
src/embeddings.py
CHANGED
|
@@ -23,10 +23,10 @@ print(f"✅ Using Hugging Face cache at {CACHE_DIR}")
|
|
| 23 |
# ----------------------------
|
| 24 |
try:
|
| 25 |
_model = SentenceTransformer(
|
| 26 |
-
"intfloat/e5-
|
| 27 |
cache_folder=CACHE_DIR
|
| 28 |
)
|
| 29 |
-
print("✅ Loaded model: intfloat/e5-
|
| 30 |
except Exception as e:
|
| 31 |
print(f"⚠️ Model load failed ({e}), falling back to MiniLM.")
|
| 32 |
_model = SentenceTransformer(
|
|
|
|
| 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")
|
| 30 |
except Exception as e:
|
| 31 |
print(f"⚠️ Model load failed ({e}), falling back to MiniLM.")
|
| 32 |
_model = SentenceTransformer(
|