Spaces:
Runtime error
Runtime error
Adrien
commited on
Commit
·
0090181
1
Parent(s):
cc3f1e1
fix: skipping reranking because HFApi is too slow
Browse files
rag_demo/rag/retriever.py
CHANGED
|
@@ -135,7 +135,9 @@ class RAGPipeline:
|
|
| 135 |
logger.info(f"Query type: {query_type}")
|
| 136 |
if query_type == "Sources_needed":
|
| 137 |
docs = self.search(query, k=10)
|
| 138 |
-
|
|
|
|
|
|
|
| 139 |
else:
|
| 140 |
reranked_docs = []
|
| 141 |
|
|
|
|
| 135 |
logger.info(f"Query type: {query_type}")
|
| 136 |
if query_type == "Sources_needed":
|
| 137 |
docs = self.search(query, k=10)
|
| 138 |
+
# disabling reranking for now because HFApi is too slow
|
| 139 |
+
# reranked_docs = self.rerank(query, docs, keep_top_k=10)
|
| 140 |
+
reranked_docs = docs
|
| 141 |
else:
|
| 142 |
reranked_docs = []
|
| 143 |
|