init vars

pages/1_Disentanglement.py

@@ -42,6 +42,10 @@ concepts = './data/concepts.txt'
 with open(concepts) as f:
     labels = [line.strip() for line in f.readlines()]
 
+if 'image_id' not in st.session_state:
+    st.session_state.image_id = 0
+if 'concept_id' not in st.session_state:
+    st.session_state.concept_id = ''
 # def on_change_random_input():
 #     st.session_state.image_id = st.session_state.image_id