Update app.py

app.py

@@ -98,11 +98,11 @@ def decompose_cavity(pred, th2=0.7, amin=10):
         xi, yi = X[b], Y[b]
         img[xi, yi] = pred[xi, yi]
 
-        # THRESHOLDING #2
-        if not (img > th2).any(): continue
+        # # THRESHOLDING #2
+        # if not (img > th2).any(): continue
 
-        # MINIMAL AREA
-        if np.sum(img) <= amin: continue
+        # # MINIMAL AREA
+        # if np.sum(img) <= amin: continue
 
         cavities.append(img)
 
@@ -114,6 +114,7 @@ def decompose_cavity(pred, th2=0.7, amin=10):
 bordersize = 0.6
 _, col, _ = st.columns([bordersize, 3, bordersize])
 
+if os.path.exists("pred.npy"): os.system("rm pred.npy")
 # os.system("rm -r predictions")
 # os.system("rm predictions.zip Views")
 # os.system("mkdir -p predictions")
@@ -174,12 +175,12 @@ if uploaded_file is not None:
             pred = np.rot90(pred, -j)
             y_pred += pred / 4
 
-        # np.save("pred.npy", y_pred)
+        np.save("pred.npy", y_pred)
 
-    # try: y_pred = np.load("pred.npy")
-    # except: y_pred = np.zeros((128,128))
-    try: y_pred
+    try: y_pred = np.load("pred.npy")
     except: y_pred = np.zeros((128,128))
+    # try: y_pred
+    # except: y_pred = np.zeros((128,128))
     y_pred_th = np.where(y_pred > threshold, y_pred, 0)
     # np.save("thresh.npy", y_pred)