singhankur01 commited on
Commit
fb34531
·
verified ·
1 Parent(s): f22caa3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -238,7 +238,7 @@ async def run_hackrx(req: RunRequest):
238
  # end_time2 = time.time() - start_time2
239
  # print(f"vector done: {end_time2}")
240
  # dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
241
- dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 16 ,"lambda_mult": 0.65} )
242
  # dense_retriever = vectorstore.as_retriever(search_type="similarity" ,search_kwargs={"k": 11} )
243
 
244
 
@@ -246,7 +246,7 @@ async def run_hackrx(req: RunRequest):
246
  keyword_retriever = BM25Retriever.from_documents(chunks)
247
  keyword_retriever.k = 11
248
  # dense_retriever = Chroma.from_documents(documents=chunks, embedding=ml_models["embedder"]).as_retriever()
249
- ensemble_retriever = EnsembleRetriever(retrievers=[keyword_retriever, dense_retriever], weights=[0.4, 0.6],search_kwargs={"k": 16})
250
  ### to make it faster we are now using our built reranker thats why commenting the code below
251
  # compression_retriever = ContextualCompressionRetriever(
252
  # base_retriever=ensemble_retriever, base_compressor=ml_models["reranker_compressor"]
 
238
  # end_time2 = time.time() - start_time2
239
  # print(f"vector done: {end_time2}")
240
  # dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
241
+ dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 16 ,"lambda_mult": 0.7} )
242
  # dense_retriever = vectorstore.as_retriever(search_type="similarity" ,search_kwargs={"k": 11} )
243
 
244
 
 
246
  keyword_retriever = BM25Retriever.from_documents(chunks)
247
  keyword_retriever.k = 11
248
  # dense_retriever = Chroma.from_documents(documents=chunks, embedding=ml_models["embedder"]).as_retriever()
249
+ ensemble_retriever = EnsembleRetriever(retrievers=[keyword_retriever, dense_retriever], weights=[0.3, 0.7],search_kwargs={"k": 16})
250
  ### to make it faster we are now using our built reranker thats why commenting the code below
251
  # compression_retriever = ContextualCompressionRetriever(
252
  # base_retriever=ensemble_retriever, base_compressor=ml_models["reranker_compressor"]