singhankur01 commited on
Commit
a68e2d0
·
verified ·
1 Parent(s): 66de3d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -162,14 +162,14 @@ 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": 12 ,"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 = 8
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": 12})
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"]
 
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": 10 ,"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 = 6
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.4, 0.6],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"]