singhankur01 commited on
Commit
860b3a8
·
verified ·
1 Parent(s): 0a7f715

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -161,12 +161,13 @@ async def run_hackrx(req: RunRequest):
161
  print(f"chunking done: {end_time}")
162
  if not chunks:
163
  return JSONResponse({"error": "No documents could be processed."}, status_code=400)
164
-
165
  vectorstore = await FAISS.afrom_documents(
166
  documents=chunks,
167
  embedding=ml_models["embedder"]
168
  )
169
-
 
170
  # dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
171
  dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 12 ,"lambda_mult": 0.5} )
172
 
 
161
  print(f"chunking done: {end_time}")
162
  if not chunks:
163
  return JSONResponse({"error": "No documents could be processed."}, status_code=400)
164
+ start_time2 = time.time()
165
  vectorstore = await FAISS.afrom_documents(
166
  documents=chunks,
167
  embedding=ml_models["embedder"]
168
  )
169
+ end_time2 = time.time() - start_time2
170
+ print(f"vector done: {end_time2}")
171
  # dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
172
  dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 12 ,"lambda_mult": 0.5} )
173