add colors

app.py

@@ -160,11 +160,13 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
                 # get unique RGB colors from this segmentation map
                 st.image(Image.fromarray(real_seg), width=512)
                 print(real_seg.shape)
+                print(real_seg)
                 unique_colors = np.unique(real_seg.reshape(-1, real_seg.shape[2]), axis=0)
+                print(unique_colors)
                 st.session_state['unique_colors'] = unique_colors
                 # multiselect to choose multiple colors
                 chosen_colors = st.multiselect(
-                    "Choose colors", list(unique_colors), format_func=map_colors, key='chosen_colors'
+                    "Choose colors", list(unique_colors), key='unique_colors'
                 )
                 print(chosen_colors)