Update reranker.py

reranker.py

@@ -10,9 +10,8 @@ def reranker(query, matches):
         pairs.append([query, match["metadata"]["text"]])
     
     model = CrossEncoder('cross-encoder/ms-marco-TinyBERT-L-2-v2', max_length = 512)
-    print("Pairs variable:", pairs)
 
     scores = model.predict(pairs)
-    top_indices = np.argsort(scores)[::-5]
+    top_indices = np.argsort(scores)[::-4]
     top_results = ["Class: " + matches[index]["metadata"]["text"] for index in top_indices]
     return top_results