Spaces:
Running
on
A10G
Running
on
A10G
BertChristiaens
commited on
Commit
•
61a9d1f
1
Parent(s):
5e179d4
change to tuple
Browse files
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(
|