Update app.py

app.py

@@ -96,7 +96,7 @@ def cut_n_predict(data, wcs, scale):
 
 # Define function to decompose prediction into individual cavities
 @st.cache
-def decompose_cavity(pred, th2=0.7, amin=10):
+def decompose_cavity(pred, th2=0.7, amin=6):
     X, Y = pred.nonzero()
     data = np.array([X,Y]).reshape(2, -1)
 
@@ -113,8 +113,8 @@ 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