prjection as state id

pages/1_Omniart_Disentanglement.py

@@ -46,6 +46,8 @@ with open(concepts) as f:
 
 if 'image_id' not in st.session_state:
     st.session_state.image_id = 0
+if 'projection' not in st.session_state:
+    st.session_state.projection = False
 if 'concept_id' not in st.session_state:
     st.session_state.concept_id = 'Abstract'
 if 'space_id' not in st.session_state:
@@ -132,10 +134,10 @@ with input_col_2:
             image_id = random.randint(0, 50000)
             st.session_state.image_id = image_id
             chosen_image_id_input.number_input('Image ID:', format='%d', step=1, value=st.session_state.image_id)
-            projection = False
+            st.session_state.projection = False
             
             if projection_id:
-                projection = True
+                st.session_state.projection = True
             
         if choose_image_button:
             image_id = int(image_id)
@@ -167,7 +169,7 @@ if st.session_state.space_id == 'Z':
 else:
     original_image_vec = annotations['w_vectors'][st.session_state.image_id]
 
-if projection:
+if st.session_state.projection:
     original_image_vec = original_image_vec - np.dot(original_image_vec.T, separation_vector) * separation_vector
 
 img = generate_original_image(original_image_vec, model, latent_space=st.session_state.space_id)