commit

app.py

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