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