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