singhankur01 commited on
Commit
5a29607
·
verified ·
1 Parent(s): 5b933bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -204,7 +204,7 @@ async def run_hackrx(req: RunRequest):
204
  # joined_context = "\n\n".join(top_chunks)
205
  # retrieved_chunks_all.append(joined_context)
206
  async def async_retrieve_and_rerank(question: str, q_idx: int):
207
- docs = ensemble_retriever.get_relevant_documents(question)
208
  doc_texts = [doc.page_content for doc in docs]
209
  doc_embeddings = ml_models["embedder"].embed_documents(doc_texts)
210
  sims = cosine_similarity([question_embeddings[q_idx]], doc_embeddings)[0]
 
204
  # joined_context = "\n\n".join(top_chunks)
205
  # retrieved_chunks_all.append(joined_context)
206
  async def async_retrieve_and_rerank(question: str, q_idx: int):
207
+ docs = await ensemble_retriever.ainvoke(question)
208
  doc_texts = [doc.page_content for doc in docs]
209
  doc_embeddings = ml_models["embedder"].embed_documents(doc_texts)
210
  sims = cosine_similarity([question_embeddings[q_idx]], doc_embeddings)[0]