singhankur01 commited on
Commit
9bc6cab
·
verified ·
1 Parent(s): 0fb8c62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -69,7 +69,7 @@ async def lifespan(app: FastAPI):
69
  )
70
  cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
71
  # cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
72
- ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=5)
73
  ml_models["llm"] = ChatGoogleGenerativeAI(
74
  model="gemini-1.5-pro",
75
  api_key=GOOGLE_API_KEY,
@@ -163,7 +163,7 @@ async def run_hackrx(req: RunRequest):
163
  )
164
 
165
  # dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
166
- dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8 ,"lambda_mult": 0.5})
167
 
168
 
169
  # Create retrievers using the pre-loaded models from our ml_models dictionary
 
69
  )
70
  cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
71
  # cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
72
+ ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=7)
73
  ml_models["llm"] = ChatGoogleGenerativeAI(
74
  model="gemini-1.5-pro",
75
  api_key=GOOGLE_API_KEY,
 
163
  )
164
 
165
  # dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
166
+ dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 9 ,"lambda_mult": 0.5})
167
 
168
 
169
  # Create retrievers using the pre-loaded models from our ml_models dictionary