Hugging Face's logo

Spaces:
olyandrevn
/
rag_homework
Sleeping

App Files Community
root commited on
Commit
fe8b339
·
1 Parent(s): 144bde6

add app

Browse files
Files changed (1) hide show
  1. backend/semantic_search.py +1 -1
backend/semantic_search.py CHANGED
@@ -30,7 +30,7 @@ def retrieve(query, n):
30
  def rerank(query, documents, k):
31
  query_doc_pairs = [[query, doc] for doc in documents]
32
  similarity_scores = reranker.predict(query_doc_pairs)
33
- sim_scores_argsort = reversed(np.argsort(similarity_scores))
34
 
35
  rerank_documents = []
36
 
 
30
  def rerank(query, documents, k):
31
  query_doc_pairs = [[query, doc] for doc in documents]
32
  similarity_scores = reranker.predict(query_doc_pairs)
33
+ sim_scores_argsort = np.argsort(similarity_scores)[::-1]
34
 
35
  rerank_documents = []
36