singhankur01 commited on
Commit
4214e1d
·
verified ·
1 Parent(s): 5c200a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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": 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])
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
 
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