Update app.py

app.py

@@ -77,15 +77,15 @@ def aggregate(list_of_hits):
     for doc in list_of_hits[0]:
         rank2doc[doc["rank"]] = doc
     return [rank2doc[rank] for rank in y_optimal]
-    
+
+aggregated_ranking = aggregate(query2outputs[search_query])
 
 if search_query or button_clicked:
-    
+
     num_results = None
     t_0 = time.time()
-    search_results = query2outputs[search_query][0] # first from the 20
-
-    print(len(search_results), type(search_results[0]))
+    # search_results = query2outputs[search_query][0] # first from the 20
+    search_results = aggregated_ranking
 
     st.write(
         f'<p align=\"right\" style=\"color:grey;\"> Before aggregation for query [{search_query}] ms</p>', unsafe_allow_html=True)