Update app.py

app.py

@@ -110,7 +110,7 @@ if uploaded_file is not None:
         wcs = WCS(hdul[0].header)
     
     # Make four columns for buttons
-    _, col1, col2, col3, col4, _ = st.columns([bordersize,0.75,0.75,0.75,0.75,bordersize])
+    _, col1, col2, col3, col4, col5, col6, _ = st.columns([bordersize,0.5,0.5,0.5,0.5,0.5,0.5,bordersize])
     col1.subheader("Input image")
     col3.subheader("Prediction")
     
@@ -121,11 +121,11 @@ if uploaded_file is not None:
         scale = st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden")
         scale = int(scale.split("x")[0]) // 128
     
-    with col2:
+    with col3:
         st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
         detect = st.button('Detect cavities')
 
-    with col3:
+    with col5:
         decompose = st.button('Docompose cavities')
         
     # Make two columns for plots
@@ -146,7 +146,7 @@ if uploaded_file is not None:
             y_pred += pred / 4
 
         with colB:
-            threshold = st.slider("", 0, 1, 0.5, 0.05, label_visibility="hidden")
+            threshold = st.slider("", 0.0, 1.0, 0.5, 0.05, label_visibility="hidden")
 
         # Thresholding
         y_pred = np.where(y_pred > threshold, y_pred, 0)