adjusted K-params
Browse files
app.py
CHANGED
@@ -345,7 +345,7 @@ class BSIChatbot:
|
|
345 |
if bm25_retriever is None:
|
346 |
bm25_retriever = BM25Retriever.from_documents(allDocs)
|
347 |
#TODO!
|
348 |
-
retriever_k=
|
349 |
bm25_retriever.k= retriever_k
|
350 |
vectordb = vectorstore.as_retriever(search_kwargs={"k":retriever_k})
|
351 |
ensemble_retriever = EnsembleRetriever(retrievers=[bm25_retriever, vectordb], weights=[0.5, 0.5])
|
@@ -380,7 +380,7 @@ class BSIChatbot:
|
|
380 |
#new print(retrieved_chunks_text)
|
381 |
with torch.no_grad():
|
382 |
print("reranking chunks..")
|
383 |
-
retrieved_chunks_text = rerankingModel.rerank(query, retrieved_chunks_text, k=
|
384 |
#newprint("DBG:retrieved_chunks_text after rerank")
|
385 |
#newprint(type(retrieved_chunks_text))
|
386 |
#newprint(retrieved_chunks_text)
|
|
|
345 |
if bm25_retriever is None:
|
346 |
bm25_retriever = BM25Retriever.from_documents(allDocs)
|
347 |
#TODO!
|
348 |
+
retriever_k=25
|
349 |
bm25_retriever.k= retriever_k
|
350 |
vectordb = vectorstore.as_retriever(search_kwargs={"k":retriever_k})
|
351 |
ensemble_retriever = EnsembleRetriever(retrievers=[bm25_retriever, vectordb], weights=[0.5, 0.5])
|
|
|
380 |
#new print(retrieved_chunks_text)
|
381 |
with torch.no_grad():
|
382 |
print("reranking chunks..")
|
383 |
+
retrieved_chunks_text = rerankingModel.rerank(query, retrieved_chunks_text, k=20)
|
384 |
#newprint("DBG:retrieved_chunks_text after rerank")
|
385 |
#newprint(type(retrieved_chunks_text))
|
386 |
#newprint(retrieved_chunks_text)
|