Update app.py

app.py

@@ -92,11 +92,11 @@ def search(query):
         hits[idx]['cross-score'] = cross_scores[idx]
 
     # Output of top-N hits from bi-encoder
-    st.write("\n-------------------------\n")
-    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", " ")))
+    #st.write("\n-------------------------\n")
+    #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")
@@ -110,9 +110,9 @@ def search(query):
     if q not in hit_res:
         hit_res.append(q)
     for res in hit_res[0:maxtags_sidebar]:
-      keywords = custom_kw_extractor.extract_keywords(res)
-      for kw in keywords:
-        print(kw)
+        keywords = custom_kw_extractor.extract_keywords(res)
+        for kw in keywords:
+            st.write(kw)
 
 st.write("## Results:")
 if st.button('Generated Expansion'):