bharatverse11 commited on
Commit
6755335
·
verified ·
1 Parent(s): f7df6ad

Update reranker.py

Browse files
Files changed (1) hide show
  1. reranker.py +3 -2
reranker.py CHANGED
@@ -12,8 +12,9 @@ pip install sentence-transformers
12
  from sentence_transformers import CrossEncoder
13
 
14
  # Best model for semantic relevance matching
15
- # Best model for semantic relevance matching (Upgraded to L-12 for maximum accuracy)
16
- RERANK_MODEL = "cross-encoder/ms-marco-MiniLM-L-12-v2"
 
17
 
18
  class Reranker:
19
 
 
12
  from sentence_transformers import CrossEncoder
13
 
14
  # Best model for semantic relevance matching
15
+ # UPGRADE: Using BAAI/bge-reranker-base for State-of-the-Art Accuracy
16
+ # This model outperforms MS-MARCO MiniLM significantly
17
+ RERANK_MODEL = "BAAI/bge-reranker-base"
18
 
19
  class Reranker:
20