singhankur01 commited on
Commit
0ef0743
·
verified ·
1 Parent(s): f8aebeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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": 5})
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 = 4
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.2, 0.8])
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"]