Update app.py

app.py

@@ -110,6 +110,8 @@ def plot_outlines(img, masks):
     buf.seek(0)
     pil_img = Image.open(buf)
 
+    plt.close(fig)
+
     return pil_img
 
 def plot_overlay(img, masks):
@@ -131,7 +133,7 @@ def plot_overlay(img, masks):
 
 def normalize99(img):
     X = img.copy()
-    X = (X - np.percentile(X, 1)) / (np.percentile(X, 99) - np.percentile(X, 1))
+    X = (X - np.percentile(X, 1)) / (1e-10 + np.percentile(X, 99) - np.percentile(X, 1))
     return X
 
 def image_resize(img, resize=400):