Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -69,7 +69,7 @@ async def lifespan(app: FastAPI):
|
|
| 69 |
)
|
| 70 |
cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
|
| 71 |
# cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
|
| 72 |
-
ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=
|
| 73 |
ml_models["llm"] = ChatGoogleGenerativeAI(
|
| 74 |
model="gemini-1.5-pro",
|
| 75 |
api_key=GOOGLE_API_KEY,
|
|
@@ -163,7 +163,7 @@ async def run_hackrx(req: RunRequest):
|
|
| 163 |
)
|
| 164 |
|
| 165 |
# dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
|
| 166 |
-
dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k":
|
| 167 |
|
| 168 |
|
| 169 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|
|
|
|
| 69 |
)
|
| 70 |
cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
|
| 71 |
# cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
|
| 72 |
+
ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=7)
|
| 73 |
ml_models["llm"] = ChatGoogleGenerativeAI(
|
| 74 |
model="gemini-1.5-pro",
|
| 75 |
api_key=GOOGLE_API_KEY,
|
|
|
|
| 163 |
)
|
| 164 |
|
| 165 |
# dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
|
| 166 |
+
dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 9 ,"lambda_mult": 0.5})
|
| 167 |
|
| 168 |
|
| 169 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|