Spaces:
Sleeping
Sleeping
Update advanced_rag.py
Browse files- advanced_rag.py +4 -3
advanced_rag.py
CHANGED
|
@@ -26,10 +26,11 @@ class AdvancedRAG:
|
|
| 26 |
self.embedding_service = embedding_service
|
| 27 |
self.qdrant_service = qdrant_service
|
| 28 |
|
| 29 |
-
# Initialize Cross-Encoder for reranking (
|
| 30 |
print("Loading Cross-Encoder model for reranking...")
|
| 31 |
-
|
| 32 |
-
|
|
|
|
| 33 |
|
| 34 |
def expand_query_llm(
|
| 35 |
self,
|
|
|
|
| 26 |
self.embedding_service = embedding_service
|
| 27 |
self.qdrant_service = qdrant_service
|
| 28 |
|
| 29 |
+
# Initialize Cross-Encoder for reranking (multilingual for Vietnamese support)
|
| 30 |
print("Loading Cross-Encoder model for reranking...")
|
| 31 |
+
# Use multilingual model instead of English-only ms-marco
|
| 32 |
+
self.cross_encoder = CrossEncoder('cross-encoder/mmarco-mMiniLMv2-L12-H384-v1')
|
| 33 |
+
print("✓ Cross-Encoder loaded (multilingual)")
|
| 34 |
|
| 35 |
def expand_query_llm(
|
| 36 |
self,
|