Spaces:

z00mP
/

Simple-RAG-solution

Sleeping

z00mP commited on Feb 21, 2024

Commit

fc8de1e

1 Parent(s): 97723c6

fix cross-encoder launch

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ def bot(history, api_kind, chunk_table, embedding_model, llm_model, cross_encode
     #documents = retrieve(query, TOP_K)
     documents = retrieve(query, top_k_param, chunk_table, embedding_model)
     if cross_encoder != "None" and len(documents) > 1:
-        documents = rerank_documents(query, documents, query, top_k_rerank=rerank_topk)
         #"cross-encoder/ms-marco-MiniLM-L-6-v2"

     #documents = retrieve(query, TOP_K)
     documents = retrieve(query, top_k_param, chunk_table, embedding_model)
     if cross_encoder != "None" and len(documents) > 1:
+        documents = rerank_documents(cross_encoder, documents, query, top_k_rerank=rerank_topk)
         #"cross-encoder/ms-marco-MiniLM-L-6-v2"