commit

app.py

@@ -83,11 +83,10 @@ if file_name is not None:
     for label, color in enumerate(palette):
         color_seg[seg == label, :] = color
         
-        color_seg = color_seg[..., ::-1]
+    color_seg = color_seg[..., ::-1]
         
-        img = np.array(image) * 0.5 + color_seg * 0.5
-        img = img.astype(np.uint8)
-        
-        plt.figure(figsize=(15, 10))
-        plt.title("Image with Segmentation")
-        st.pyplot(plt.gcf())
+    img = np.array(image) * 0.5 + color_seg * 0.5
+    img = img.astype(np.uint8)
+    plt.figure(figsize=(15, 10))
+    plt.title("Image with Segmentation")
+    st.pyplot(plt.gcf())