bug fix for finish button

pages/Ranking.py

@@ -114,7 +114,7 @@ class RankingApp:
         # save the current progress to session state
         if 'progress' not in st.session_state:
             st.session_state.progress = {}
-        # print('current progress: ', st.session_state.progress)
+        print('current progress: ', st.session_state.progress)
 
         prompt_tags, tag, prompt_id, items = self.sidebar()
         batch_num = len(items) // self.batch_size
@@ -183,8 +183,7 @@ class RankingApp:
         ranking_dataset.push_to_hub('MAPS-research/GEMRec-Ranking', split='train')
 
         if progress == 'finished':
-            st.session_state.progress['prompt_id'] = 'finished'
-            st.experimental_rerun()
+            st.session_state.progress[prompt_id] = 'finished'
         else:
             st.session_state.counter[prompt_id] += 1