cols changed

pages/1_Disentanglement.py

@@ -86,8 +86,10 @@ st.subheader('Concept vector')
 # perform attack container
 # header_col_1, header_col_2, header_col_3, header_col_4, header_col_5 = st.columns([1,1,1,1,1])
 # output_col_1, output_col_2, output_col_3, output_col_4, output_col_5 = st.columns([1,1,1,1,1])
-header_col_1 = st.columns([1])
-output_col_1 = st.columns([1])
+header_col_1 = st.columns(1)
+output_col_1 = st.columns(1)
+
+st.subheader('Derivations along the concept vector')
 
 # prediction error container
 error_container = st.empty()