minhvtt commited on
Commit
5838fbc
·
verified ·
1 Parent(s): 1009897

Update advanced_rag.py

Browse files
Files changed (1) hide show
  1. 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 (state-of-the-art)
30
  print("Loading Cross-Encoder model for reranking...")
31
- self.cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
32
- print("✓ Cross-Encoder loaded")
 
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,