helal94hb1 commited on
Commit
9d90187
·
1 Parent(s): 8595342

fix: new reranker2

Browse files
Files changed (1) hide show
  1. app/core/config.py +1 -1
app/core/config.py CHANGED
@@ -26,7 +26,7 @@ class Settings(BaseSettings):
26
  S3_RERANKER_URL: Optional[str] = None
27
 
28
  RERANKER_MODEL_PATH: str = "data/best_expert_judge_cross_encoder.pt" # Or the exact name of your saved .pt file
29
- RERANKER_MODEL_NAME: str = "bert-base-uncased" # The base model used in your training script
30
  RERANKER_SCORE_THRESHOLD: float = 0.3
31
 
32
  SEQUENCE_EXPANSION_THRESHOLD: float =0.68
 
26
  S3_RERANKER_URL: Optional[str] = None
27
 
28
  RERANKER_MODEL_PATH: str = "data/best_expert_judge_cross_encoder.pt" # Or the exact name of your saved .pt file
29
+ RERANKER_MODEL_NAME: str = "BAAI/bge-reranker-base" # The base model used in your training script
30
  RERANKER_SCORE_THRESHOLD: float = 0.3
31
 
32
  SEQUENCE_EXPANSION_THRESHOLD: float =0.68