Spaces:
Sleeping
Sleeping
root
commited on
Commit
·
fe8b339
1
Parent(s):
144bde6
add app
Browse files
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 =
|
| 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 |
|