Update app.py

app.py

@@ -85,14 +85,14 @@ def search(query):
 
     # Output of top-N hits from bi-encoder
     st.write("\n-------------------------\n")
-    st.write("Top-N Bi-Encoder Retrieval hits")
+    st.subheader("Top-N Bi-Encoder Retrieval hits")
     hits = sorted(hits, key=lambda x: x['score'], reverse=True)
     for hit in hits[0:maxtags_sidebar]:
         st.write("\t{:.3f}\t{}".format(hit['score'], passages[hit['corpus_id']].replace("\n", " ")))
 
     # Output of top-N hits from re-ranker
     st.write("\n-------------------------\n")
-    st.write("Top-N Cross-Encoder Re-ranker hits")
+    st.subheader("Top-N Cross-Encoder Re-ranker hits")
     hits = sorted(hits, key=lambda x: x['cross-score'], reverse=True)
     for hit in hits[0:maxtags_sidebar]:
         st.write("\t{:.3f}\t{}".format(hit['cross-score'], passages[hit['corpus_id']].replace("\n", " ")))
@@ -100,4 +100,4 @@ def search(query):
 st.write("## Results:")
 if st.button('Generated Expansion'):
     out = search(query = user_query)
-    st.success(out)
+    #st.success(out)