Update app.py

app.py

@@ -236,16 +236,14 @@ def re_rank_candidates(query, candidates, method):
     return df
 
 col1, col2 = st.columns(2)
-col1.subheader('generate expansion')
-col2.subheader('rereank by gms')
 if st.button('Generated Expansion'): 
     candidates = generate_query_expansion_candidates(query = user_query)
-    st.write("Raw Candidates:")
+    col1.subheader('Raw Candidates:')
     df = re_rank_candidates(user_query, candidates, method='cross_encoder')
     result = list(df['query'][:maxtags_sidebar])
     st.write(result)
 
-    st.write("Rerank By GMS:")
+    col2.subheader('Rerank By GMS:')
     df = re_rank_candidates(user_query, candidates, method='gms')
     result = df[['query', 'gms']][:maxtags_sidebar]
-    st.table(result)
+    st.json(result.to_dict('list'))