Update app.py

app.py

@@ -115,11 +115,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)
 
@@ -231,5 +231,6 @@ if uploaded_file is not None:
         shutil.make_archive("predictions", 'zip', "predictions")
         with open('predictions.zip', 'rb') as f:
             res = f.read()
+        st.markdown("")
         # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
         download = st.download_button(label="Download", data=res, file_name='prediction.zip', mime="application/octet-stream")