singhankur01 commited on
Commit
9ab78f8
·
verified ·
1 Parent(s): cb0dae8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -62,8 +62,8 @@ async def lifespan(app: FastAPI):
62
 
63
  # Load models into the shared dictionary
64
  ml_models["embedder"] = HuggingFaceEmbeddings(
65
- # model_name="BAAI/bge-base-en-v1.5", #better but lil slower
66
- model_name="intfloat/e5-large-v2", #lil faster
67
  encode_kwargs={
68
  "batch_size": 64,
69
  # "normalize_embeddings": True
@@ -167,7 +167,7 @@ async def run_hackrx(req: RunRequest):
167
  )
168
 
169
  # dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
170
- dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 7 ,"lambda_mult": 0.5})
171
 
172
 
173
  # Create retrievers using the pre-loaded models from our ml_models dictionary
 
62
 
63
  # Load models into the shared dictionary
64
  ml_models["embedder"] = HuggingFaceEmbeddings(
65
+ model_name="BAAI/bge-base-en-v1.5", #better but lil slower
66
+ # model_name="intfloat/e5-large-v2", #lil faster
67
  encode_kwargs={
68
  "batch_size": 64,
69
  # "normalize_embeddings": True
 
167
  )
168
 
169
  # dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
170
+ dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8 ,"lambda_mult": 0.5})
171
 
172
 
173
  # Create retrievers using the pre-loaded models from our ml_models dictionary