Spaces:
Sleeping
Sleeping
Commit
·
9d90187
1
Parent(s):
8595342
fix: new reranker2
Browse files- 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 = "
|
| 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
|