Update app.py

app.py

@@ -203,7 +203,8 @@ if uploaded_file is not None:
     image = np.log10(data+1)
     plot_image(image, scale)
     
-    if detect or threshold:
+    # if detect or threshold:
+    if st.session_state.get("detect", True)
         y_pred, wcs = cut_n_predict(data, wcs, scale)
         
         y_pred_th = np.where(y_pred > threshold, y_pred, 0)