important nodes list

pages/1_Disentanglement.py

@@ -100,10 +100,10 @@ smoothgrad_col_1, smoothgrad_col_2, smoothgrad_col_3, smoothgrad_col_4, smoothgr
 
 # ---------------------------- DISPLAY COL 1 ROW 1 ------------------------------
 with output_col_1:
-    separation_vector, number_important_features = get_separation_space(concept_id, annotations, ann_df)
+    separation_vector, number_important_features, imp_nodes = get_separation_space(concept_id, annotations, ann_df)
     # st.write(f'Class ID {input_id} - {input_label}: {pred_prob*100:.3f}% confidence')
     st.write('Concept vector', separation_vector)
-    header_col_1.write(f'Concept {concept_id} - Number of relevant nodes: {number_important_features}')
+    header_col_1.write(f'Concept {concept_id} - Number of relevant nodes: {number_important_features} - Nodes {",".join(list(imp_nodes))}')
 
 # ----------------------------- INPUT column 2 & 3 ----------------------------        
 with input_col_2: