BertChristiaens commited on
Commit
61a9d1f
1 Parent(s): 5e179d4

change to tuple

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -160,9 +160,9 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
160
  # get unique RGB colors from this segmentation map
161
  st.image(Image.fromarray(real_seg), width=512)
162
  print(real_seg.shape)
163
- print(real_seg)
164
  unique_colors = np.unique(real_seg.reshape(-1, real_seg.shape[2]), axis=0)
165
  print(unique_colors)
 
166
  st.session_state['unique_colors'] = unique_colors
167
  # multiselect to choose multiple colors
168
  chosen_colors = st.multiselect(
 
160
  # get unique RGB colors from this segmentation map
161
  st.image(Image.fromarray(real_seg), width=512)
162
  print(real_seg.shape)
 
163
  unique_colors = np.unique(real_seg.reshape(-1, real_seg.shape[2]), axis=0)
164
  print(unique_colors)
165
+ unique_colors = (tuple(color) for color in unique_colors)
166
  st.session_state['unique_colors'] = unique_colors
167
  # multiselect to choose multiple colors
168
  chosen_colors = st.multiselect(