bbb

pages/3_Oxford_Vases_Disentanglement.py

@@ -147,7 +147,8 @@ img = generate_original_image(original_image_vec, model, latent_space=st.session
 cols = list(ann_df.columns)
 cols.remove('Unnamed: 0')
 cols.remove('ID')
-top_pred = ann_df.loc[st.session_state.image_id, cols].idxmax()
+print(ann_df.loc[st.session_state.image_id, cols])
+top_pred = ann_df.loc[st.session_state.image_id, cols].astype(float).idxmax()
 # input_image = original_image_dict['image']
 # input_label = original_image_dict['label']
 # input_id = original_image_dict['id']