bug fix

backend/disentangle_concepts.py

@@ -43,6 +43,7 @@ def get_separation_space(type_bin, annotations, df, samples=200, method='LR', C=
     else:
         print('Error: type_bin must be either a string or a list of strings of len 2')
         return
+    
     X = X.reshape((2*samples, 512))
     y = np.array([1]*samples + [0]*samples)
     x_train, x_val, y_train, y_val = train_test_split(X, y, test_size=0.2)

pages/3_Oxford_Vases_Disentanglement.py

@@ -56,7 +56,7 @@ with input_col_1:
         concept_ids = st.multiselect('Concepts:', tuple(labels), max_selections=2, default=['AMPHORA', 'CHALICE'])
 
         st.write('**Choose a latent space to disentangle**')
-        space_id = st.selectbox('Space:', tuple(['Z', 'W']), default='W')
+        space_id = st.selectbox('Space:', tuple(['W', 'Z']))
 
         choose_text_button = st.form_submit_button('Choose the defined concept and space to disentangle')