Update app.py

app.py

@@ -250,7 +250,7 @@ if st.button('Generated Expansion'):
     with col2:
         st.subheader('Sorted Query Candidates')
         df2 = re_rank_candidates(user_query, candidates, method='gms')
-        result_rank=list(df2['query', 'gms'][:maxtags_sidebar])
+        result_rank=list(df2[['query', 'gms']][:maxtags_sidebar])
 
     ## convert into dataframe
     # data_dicts = [{'query': key, **values} for key, values in candidates.items()]