Update app.py

app.py

@@ -66,38 +66,33 @@ def cut(data0, wcs0, scale=1):
 
     return data, wcs
 
-# Make two columns
-col1, col2 = st.columns(2)
-col1.subheader("Input image")
-col2.subheader("CADET prediction")
-
-with col1:
-    st.markdown(
-        """<style>[data-baseweb="select"] {margin-top: -50px;}</style>""",
-        unsafe_allow_html=True
-    )
-
-    max_scale = int(data.shape[0] // 128)
-    # scale = st.slider("Scale", 1, max_scale, 1, 1)
-    scale = int(st.selectbox('Scale:',[i+1 for i in range(max_scale)], label_visibility="hidden"))
-
-with col2:
-    button == st.button('Detect cavities', disabled=True)
-
     
 # If file is uploaded, read in the data and plot it
 if uploaded_file is not None:
     with fits.open(uploaded_file) as hdul:
         data = hdul[0].data
         wcs = WCS(hdul[0].header)
+    
+    # Make two columns
+    col1, col2 = st.columns(2)
+    col1.subheader("Input image")
+    col2.subheader("CADET prediction")
+    
+    with col1:
+        st.markdown(
+            """<style>[data-baseweb="select"] {margin-top: -50px;}</style>""",
+            unsafe_allow_html=True
+        )
+    
+        max_scale = int(data.shape[0] // 128)
+        # scale = st.slider("Scale", 1, max_scale, 1, 1)
+        scale = int(st.selectbox('Scale:',[i+1 for i in range(max_scale)], label_visibility="hidden"))
 
     # Add a slider to change the scale
     with col1:
         plot_image(np.log10(data+1), scale)
-
-    button.disabled = False
         
-    if button:
+    if st.button('Detect cavities'):
         data, wcs = cut(data, wcs, scale=scale)
 
         image = np.log10(data+1)