Update app.py

app.py

@@ -233,10 +233,9 @@ def re_rank_candidates(query, candidates, method):
     df = pd.DataFrame(data_dicts)
     return df
 
-
+candidates = generate_query_expansion_candidates(query = user_query)
 # st.write("## Raw Candidates:")
-if st.button('Generated Expansion'):
-    candidates = generate_query_expansion_candidates(query = user_query)
+if st.button('Generated Expansion'): 
     # convert into dataframe
     data_dicts = [{'query': key, **values} for key, values in candidates.items()]
     df = pd.DataFrame(data_dicts)
@@ -247,6 +246,6 @@ if st.button('Generated Expansion'):
 if st.button('Rerank By Encoder'):
     # raw_candidates = generate_query_expansion_candidates(query = user_query)
     # candidates = add_gms_score_for_candidates(raw_candidates, query_gms_dict)
-    out_res = re_rank_candidates(user_query, candidates, method='encoder')
+    df = re_rank_candidates(user_query, candidates, method='encoder')
     st.write("Reranking By Encoder: \n")
-    st.write(out_res[:maxtags_sidebar])
+    st.write(df[['query', 'cross_score', 'bi_score']][:maxtags_sidebar])