Update app.py

app.py

@@ -160,7 +160,7 @@ if uploaded_file is not None:
     col1.subheader("Input image")
     col3.subheader("Prediction")
     col5.subheader("Decomposed")
-    # col6.subheader("")
+    col6.subheader("")
     
     with col1:
         st.markdown("""<style>[data-baseweb="select"] {margin-top: -26px;}</style>""", unsafe_allow_html=True)
@@ -173,6 +173,7 @@ if uploaded_file is not None:
 
     # Threshold slider
     with col4:
+        st.markdown("")
         # st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
         threshold = st.slider("Threshold", 0.0, 1.0, 0.0, 0.05) #, label_visibility="hidden")
         
@@ -200,7 +201,8 @@ if uploaded_file is not None:
         y_pred_th = np.where(y_pred > threshold, y_pred, 0)
                 
         plot_prediction(y_pred_th)
-        
+
+        print(type(download))
         if decompose or download:
             download.disabled = False