Update app.py

app.py

@@ -173,17 +173,9 @@ def search_func(query):
     for idx in range(len(cross_scores)):
         hits[idx]['cross-score'] = cross_scores[idx]
 
-    # Output of top hits from bi-encoder
-    st.markdown("\n-------------------------\n")
-    st.subheader(f"Top 2 Bi-Encoder Retrieval hits")
-    hits = sorted(hits, key=lambda x: x['score'], reverse=True)
-
-    cross_df = display_as_table(hits, )
-    st.write(cross_df.to_html(index=False), unsafe_allow_html=True)
-
     # Output of top hits from cross encoder
     st.markdown("\n-------------------------\n")
-    st.subheader(f"Top-2 Cross-Encoder Re-ranker hits")
+    st.subheader(f"Top 2 Results")
     hits = sorted(hits, key=lambda x: x['cross-score'], reverse=True)
 
     rerank_df = display_as_table(hits, 'cross-score')