Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -157,14 +157,14 @@ async def run_hackrx(req: RunRequest):
|
|
| 157 |
embedding=ml_models["embedder"]
|
| 158 |
)
|
| 159 |
|
| 160 |
-
dense_retriever = vectorstore.as_retriever(search_kwargs={"k":
|
| 161 |
|
| 162 |
|
| 163 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|
| 164 |
keyword_retriever = BM25Retriever.from_documents(chunks)
|
| 165 |
-
keyword_retriever.k =
|
| 166 |
# dense_retriever = Chroma.from_documents(documents=chunks, embedding=ml_models["embedder"]).as_retriever()
|
| 167 |
-
ensemble_retriever = EnsembleRetriever(retrievers=[keyword_retriever, dense_retriever], weights=[0.
|
| 168 |
|
| 169 |
compression_retriever = ContextualCompressionRetriever(
|
| 170 |
base_retriever=ensemble_retriever, base_compressor=ml_models["reranker_compressor"]
|
|
|
|
| 157 |
embedding=ml_models["embedder"]
|
| 158 |
)
|
| 159 |
|
| 160 |
+
dense_retriever = vectorstore.as_retriever(search_kwargs={"k": 4})
|
| 161 |
|
| 162 |
|
| 163 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|
| 164 |
keyword_retriever = BM25Retriever.from_documents(chunks)
|
| 165 |
+
keyword_retriever.k = 3
|
| 166 |
# dense_retriever = Chroma.from_documents(documents=chunks, embedding=ml_models["embedder"]).as_retriever()
|
| 167 |
+
ensemble_retriever = EnsembleRetriever(retrievers=[keyword_retriever, dense_retriever], weights=[0.4, 0.6])
|
| 168 |
|
| 169 |
compression_retriever = ContextualCompressionRetriever(
|
| 170 |
base_retriever=ensemble_retriever, base_compressor=ml_models["reranker_compressor"]
|