Update app.py

app.py

@@ -142,7 +142,7 @@ if uploaded_file is not None:
     with colB:
         st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
         threshold = st.slider("", 0.0, 1.0, 0.4, 0.05, label_visibility="hidden")
-        plot_prediction(y_pred)
+        # plot_prediction(y_pred)
     
     if detect:
         data, wcs = cut(data, wcs, scale=scale)
@@ -156,7 +156,9 @@ if uploaded_file is not None:
             y_pred += pred / 4
 
         y_pred = np.where(y_pred > threshold, y_pred, 0)
-        
+
+        plot_prediction(y_pred)
+
         # if decompose:
         #     cavs = decompose_cavity(y_pred, )