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 |
|