Update app.py

app.py

@@ -236,12 +236,12 @@ col1, col2 = st.columns(2)
 if st.button('Generated Expansion'): 
     candidates = generate_query_expansion_candidates(query = user_query)
     with col1:
-        st.header('Raw Candidates:')
+        st.subheader('Raw Candidates:')
         candidates_rerank = re_rank_candidates(user_query, candidates, method='cross_encoder')[:maxtags_sidebar]
         result = [item[0] for item in candidates_rerank]
         st.write(result)
     with col2:
-        st.header('Rerank By GMS:')
+        st.subheader('Rerank By GMS:')
         candidates_rerank = re_rank_candidates(user_query, candidates, method='gms')[:maxtags_sidebar]
         result = [{item[0]: item[1]['gms']} for item in candidates_rerank]
-        st.json(result)
+        st.write(result)