Update app.py

app.py

@@ -26,8 +26,6 @@ uploaded_file = st.file_uploader("Choose a FITS file", type=['fits'])
 
 # Make two columns
 col1, col2 = st.columns(2)
-col1.subheader("Input image")
-col2.subheader("CADET prediction")
 
 # Define function to plot the uploaded image
 def plot_image(image_array, scale):
@@ -74,6 +72,9 @@ def cut(data0, wcs0, scale=1):
 
 # If file is uploaded, read in the data and plot it
 if uploaded_file is not None:
+    col1.subheader("Input image")
+    col2.subheader("CADET prediction")
+    
     with fits.open(uploaded_file) as hdul:
         data = hdul[0].data
         wcs = WCS(hdul[0].header)
@@ -82,12 +83,14 @@ if uploaded_file is not None:
     # Add a slider to change the scale
     with col1:
         max_scale = int(data.shape[0] // 128)
-        scale = st.slider("Scale", 1, max_scale, 1, 1)
+        # scale = st.slider("Scale", 1, max_scale, 1, 1)
+
+        scale = st.selectbox('Scale:',[i for i in range(max_scale)])
 
         plot_image(image, scale)
 
     with col2:
-        st.button('Detect cavities')                
+        st.button('Detect cavities')      
         data, wcs = cut(data, wcs, scale=scale)
 
         y_pred = 0