commit

app.py

@@ -88,7 +88,6 @@ if file_name is not None:
         img = np.array(image) * 0.5 + color_seg * 0.5
         img = img.astype(np.uint8)
         
-        plt.figure(figsize=(15, 10))
+        fig = plt.figure(figsize=(15, 10))
         plt.title("Image with Segmentation")
-        plt.imshow(img)
-        st.pyplot(img)
+        st.pyplot(fig)