Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|