Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -162,18 +162,18 @@ async def run_hackrx(req: RunRequest):
|
|
| 162 |
)
|
| 163 |
|
| 164 |
# dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
|
| 165 |
-
dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k":
|
| 166 |
|
| 167 |
|
| 168 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|
| 169 |
keyword_retriever = BM25Retriever.from_documents(chunks)
|
| 170 |
-
keyword_retriever.k =
|
| 171 |
# dense_retriever = Chroma.from_documents(documents=chunks, embedding=ml_models["embedder"]).as_retriever()
|
| 172 |
-
ensemble_retriever = EnsembleRetriever(retrievers=[keyword_retriever, dense_retriever], weights=[0.35, 0.65])
|
| 173 |
### to make it faster we are now using our built reranker thats why commenting the code below
|
| 174 |
-
compression_retriever = ContextualCompressionRetriever(
|
| 175 |
-
|
| 176 |
-
)
|
| 177 |
|
| 178 |
|
| 179 |
# Define the RAG chain using pre-loaded components
|
|
|
|
| 162 |
)
|
| 163 |
|
| 164 |
# dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
|
| 165 |
+
dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 11 ,"lambda_mult": 0.5} )
|
| 166 |
|
| 167 |
|
| 168 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|
| 169 |
keyword_retriever = BM25Retriever.from_documents(chunks)
|
| 170 |
+
keyword_retriever.k = 7
|
| 171 |
# dense_retriever = Chroma.from_documents(documents=chunks, embedding=ml_models["embedder"]).as_retriever()
|
| 172 |
+
ensemble_retriever = EnsembleRetriever(retrievers=[keyword_retriever, dense_retriever], weights=[0.35, 0.65],search_kwargs={"k": 10})
|
| 173 |
### to make it faster we are now using our built reranker thats why commenting the code below
|
| 174 |
+
# compression_retriever = ContextualCompressionRetriever(
|
| 175 |
+
# base_retriever=ensemble_retriever, base_compressor=ml_models["reranker_compressor"]
|
| 176 |
+
# )
|
| 177 |
|
| 178 |
|
| 179 |
# Define the RAG chain using pre-loaded components
|