allow selection of only one item

backend/disentangle_concepts.py

@@ -28,7 +28,9 @@ def get_separation_space(type_bin, annotations, df, samples=200, method='LR', C=
     else:
         col = 'w_vectors'
     
-    if type(type_bin) == str or len(type_bin) == 1:
+    if len(type_bin) == 1:
+        type_bin = type_bin[0]
+    if type(type_bin) == str:
         abstracts = np.array([float(ann) for ann in df[type_bin]])
         abstract_idxs = list(np.argsort(abstracts))[:samples]
         repr_idxs = list(np.argsort(abstracts))[-samples:]

pages/3_Oxford_Vases_Disentanglement.py

@@ -143,7 +143,7 @@ else:
     original_image_vec = annotations['w_vectors'][st.session_state.image_id]
 
 img = generate_original_image(original_image_vec, model, latent_space=st.session_state.space_id)
-print(ann_df.iloc[st.session_state.image_id, list(ann_df.column) - 'ID'])
+print(ann_df.iloc[st.session_state.image_id, list(ann_df.columns) - 'ID'])
 top_pred = ann_df.iloc[st.session_state.image_id, list(ann_df.columns) - 'ID'].idxmax()
 # input_image = original_image_dict['image']
 # input_label = original_image_dict['label']