Update app.py

app.py

@@ -223,7 +223,10 @@ def re_rank_candidates(query, candidates, method):
         # first sort by cross_score + bi_score
         filtered_sorted_result = sorted(filtered_sorted_by_encoder, key=lambda x: x[1]['gms'], reverse=True
         )
-    return filtered_sorted_result
+    data_dicts = [{'query': item[0], **item[1]} for item in filtered_sorted_result]
+    # Convert the list of dictionaries into a DataFrame
+    df = pd.DataFrame(data_dicts)
+    return df
 
 
 # def reranking():
@@ -256,5 +259,5 @@ if st.button('Generated Expansion'):
 if st.button('Rerank By Encoder'):
     out_res = re_rank_candidates(user_query, candidates, method='encoder')
     st.write("Reranking By Encoder: \n")
-    st.write(out_res)
+    st.write(out_res[:maxtags_sidebar])
     st.success(out_res)