Update app.py

app.py

@@ -55,22 +55,21 @@ object = torch.load(fn)
 outputs = object[2]
 query2outputs = {}
 for output in outputs:
-    print("-" * 10)
-    print(type(output[0]))
-    print(len(output[0]))
-    print("-" * 10)
     all_queries = {x['query'] for x in output}
     assert len(all_queries) == 1
     query = list(all_queries)[0]
     query2outputs[query] = [x['hits'] for x in output]
 
+print("query2outputs:", len(query2outputs))
 search_query = sorted(query2outputs)[0]
+print(search_query)
 
 if search_query or button_clicked:
     
     num_results = None
     t_0 = time.time()
     search_results = query2outputs[search_query]
+    print(search_results)
 
     st.write(
         f'<p align=\"right\" style=\"color:grey;\"> Before aggregation for query [{search_query}] ms</p>', unsafe_allow_html=True)