Fix broken rescoring

app.py

@@ -54,9 +54,10 @@ def search(query, top_k: int = 10, rescore_multiplier: int = 4):
 
     # 6. Sort the scores and return the top_k
     start_time = time.time()
-    top_k_indices = (-scores).argsort()[-top_k:]
-    top_k_scores = scores[top_k_indices]
-    top_k_titles, top_k_texts = zip(*[(title_text_dataset[idx]["title"], title_text_dataset[idx]["text"]) for idx in binary_ids[top_k_indices].tolist()])
+    indices = scores.argsort()[:top_k]
+    top_k_indices = binary_ids[indices]
+    top_k_scores = scores[indices]
+    top_k_titles, top_k_texts = zip(*[(title_text_dataset[idx]["title"], title_text_dataset[idx]["text"]) for idx in top_k_indices.tolist()])
     df = pd.DataFrame({"Score": [round(value, 2) for value in top_k_scores], "Title": top_k_titles, "Text": top_k_texts})
     sort_time = time.time() - start_time