fix session state init

app.py

@@ -1,9 +1,16 @@
 import streamlit as st
+from streamlit import session_state
 from random import choice
 from annotated_text import annotated_text
 from resources import *
 from helpers import *
 
+if "counter" not in session_state:
+    session_state.counter = 0
+
+if "augment" not in session_state:
+    session_state.augment = False
+
 base_model = "xlnet-base-cased"
 sentences = load_sentences()
 baseline_classifier = load_model(f"Dagobert42/{base_model}-biored-finetuned")
@@ -18,19 +25,19 @@ It was tagged by a model which was trained on 200 examples from the original dat
 It is very possible that there should be some mistakes.
 """)
 
-txt = sentences[st.session_state.counter]
+txt = sentences[session_state.counter]
 
-st.write("Example with data augmentation:" if st.session_state.augment else "Example without data augmentation:")
-tokens = augmented_classifier(txt) if st.session_state.augment else baseline_classifier(txt)
+st.write("Example with data augmentation:" if session_state.augment else "Example without data augmentation:")
+tokens = augmented_classifier(txt) if session_state.augment else baseline_classifier(txt)
 
 annotated_text(annotate_sentence(txt, tokens))
 st.write(annotate_sentence(txt, tokens))
 
 st.write("Now try the augmented model. Hopefully it's a bit better :)")
-st.session_state.augment = st.toggle("augmentations on" if st.session_state.augment else "augmentations off")
+session_state.augment = st.toggle("augmentations on" if session_state.augment else "augmentations off")
 
 st.write("Or load another sentence")
 def refresh_example():
-    st.session_state.counter += 1
+    session_state.counter += 1
 st.button( ":twisted_rightwards_arrows:", on_click=refresh_example)