add image

app.py

@@ -155,9 +155,11 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
             with st.spinner(text="Preparing image segmentation"):
                 image = get_image()
                 print("Preparing image segmentation")
-                real_seg = segment_image(Image.fromarray(image))
+                real_seg = np.array(segment_image(Image.fromarray(image)))
                 st.session_state['seg'] = real_seg
                 # get unique RGB colors from this segmentation map
+                st.image(Image.fromarray(real_seg), width=512)
+                print(real_seg.shape)
                 unique_colors = np.unique(real_seg.reshape(-1, real_seg.shape[2]), axis=0)
                 st.session_state['unique_colors'] = unique_colors
                 # multiselect to choose multiple colors