Update app.py

app.py

@@ -72,16 +72,18 @@ def search():
     # Sort results by the cross-encoder scores
     for idx in range(len(cross_scores)):
         hits[idx]['cross-score'] = cross_scores[idx]
-
+    
+    hits = sorted(hits, key=lambda x: x['cross-score'], reverse=True)
+    st.session_state.results = hits[:3]
     # Output of top-3 hits from re-ranker
     print("\n-------------------------\n")
     print("Top-3 Cross-Encoder Re-ranker hits")
-    hits = sorted(hits, key=lambda x: x['cross-score'], reverse=True)
-    for hit in hits[0:3]:
-        print("score: ",  round(hit['cross-score'], 3),"\n",
-              "title: ", dataset["title"][hit['corpus_id']], "\n", 
-              "substract: ", dataset["text"][hit['corpus_id']].replace("\n", " "), "\n", 
-              "link: ", dataset["url"][hit['corpus_id']],"\n")
+    st.subheader("Top-3 Search results")
+    for hit in hits[:3]:
+        st.markdown("score: " +  round(hit['cross-score'], 3),
+        st.markdown("title: " + dataset["title"][hit['corpus_id']]), 
+        st.markdown("substract: " + dataset["text"][hit['corpus_id']].replace("\n", " ")), 
+        st.markdown("link: " + dataset["url"][hit['corpus_id']])
 
 
 # search button