rename session_data to session_state

app.py

@@ -51,8 +51,8 @@ feedback_table = wandb.Table(columns=[
     'correct_label', 'correct_label_id'
 ])
 
-if 'wandb_run' not in st.session_data:
-    st.session_data.wandb_run = wandb.init(project='hydra-classifier', name='feedback-loop')
+if 'wandb_run' not in st.session_state:
+    st.session_state.wandb_run = wandb.init(project='hydra-classifier', name='feedback-loop')
 
 
 @st.cache_data
@@ -109,7 +109,7 @@ if uploaded_file:
                 st.success(f"Feedback for Image {i} submitted!")
 
     print(feedback_table)
-    run = st.session_data.wandb_run
+    run = st.session_state.wandb_run
     run.log({'feedback_table': feedback_table})
     run.finish()
-    st.session_data.wandb_run = None
+    st.session_state.wandb_run = None