Update app.py

app.py

@@ -142,9 +142,8 @@ def decompose_cavity(pred, th2=0.7, amin=6):
 #         wcs = WCS(hdul[0].header)
 #     return data, wcs
 
-def change_scale():
-    del st.session_state["threshold"]
-
+# def change_scale():
+#     del st.session_state["threshold"]
 
 
 # Use wide layout and create columns
@@ -203,7 +202,7 @@ if uploaded_file is not None:
     with col1:
         st.markdown("""<style>[data-baseweb="select"] {margin-top: -46px;}</style>""", unsafe_allow_html=True)
         max_scale = int(data.shape[0] // 128)
-        scale = st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden", on_change=change_scale)
+        scale = st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden") #, on_change=change_scale)
         scale = int(scale.split("x")[0]) // 128
     
     # Detect button
@@ -224,13 +223,13 @@ if uploaded_file is not None:
     image = np.log10(data+1)
     plot_image(image, scale)
     
-    # if detect or threshold:
-    # # if st.session_state.get("detect", True):
-    #     y_pred, wcs = cut_n_predict(data, wcs, scale)
+    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)
+        y_pred_th = np.where(y_pred > threshold, y_pred, 0)
                 
-    #     plot_prediction(y_pred_th)
+        plot_prediction(y_pred_th)
     
     #     if decompose or st.session_state.get("download", False):            
     #         image_decomposed = decompose_cavity(y_pred_th)