Spaces:
Sleeping
Sleeping
Update config/settings.py
Browse files- config/settings.py +4 -3
config/settings.py
CHANGED
|
@@ -10,8 +10,9 @@ class Config:
|
|
| 10 |
|
| 11 |
# Models - STICK WITH BGE FOR ACCURACY
|
| 12 |
EMBEDDING_MODEL = "BAAI/bge-large-en-v1.5"
|
| 13 |
-
RERANKER_MODEL = "BAAI/bge-reranker-large"
|
| 14 |
-
|
|
|
|
| 15 |
# OpenRouter
|
| 16 |
OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
|
| 17 |
OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1"
|
|
@@ -53,7 +54,7 @@ class Config:
|
|
| 53 |
# return self.PERFORMANCE_MODE in ["quality", "balanced"]
|
| 54 |
@property
|
| 55 |
def use_reranker(self):
|
| 56 |
-
return
|
| 57 |
|
| 58 |
@property
|
| 59 |
def retrieval_k(self):
|
|
|
|
| 10 |
|
| 11 |
# Models - STICK WITH BGE FOR ACCURACY
|
| 12 |
EMBEDDING_MODEL = "BAAI/bge-large-en-v1.5"
|
| 13 |
+
# RERANKER_MODEL = "BAAI/bge-reranker-large"
|
| 14 |
+
RERANKER_MODEL = "cross-encoder/ms-marco-MiniLM-L-6-v2"
|
| 15 |
+
|
| 16 |
# OpenRouter
|
| 17 |
OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
|
| 18 |
OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1"
|
|
|
|
| 54 |
# return self.PERFORMANCE_MODE in ["quality", "balanced"]
|
| 55 |
@property
|
| 56 |
def use_reranker(self):
|
| 57 |
+
return True
|
| 58 |
|
| 59 |
@property
|
| 60 |
def retrieval_k(self):
|