Update app.py

app.py

@@ -87,7 +87,7 @@ if uploaded_file is not None:
     
     with col2:
         # st.markdown("""<style>[data-baseweb="select"] {margin-top: -52px;}</style>""", unsafe_allow_html=True)
-        st.markdown("""<style>[data-baseweb="select"] {margin-top: 52px;}</style>""", unsafe_allow_html=True)
+        st.markdown("""<style>[data-baseweb="select"] {margin-top: 32px;}</style>""", unsafe_allow_html=True)
         
         max_scale = int(data.shape[0] // 128)
         # scale = st.slider("Scale", 1, max_scale, 1, 1)
@@ -97,7 +97,7 @@ if uploaded_file is not None:
         detect = st.button('Detect cavities')
     
     with col4:
-        st.markdown("""<style>[data-baseweb="select"] {margin-top: 52px;}</style>""", unsafe_allow_html=True)
+        st.markdown("""<style>[data-baseweb="select"] {margin-top: 32px;}</style>""", unsafe_allow_html=True)
         download = st.button('Download')
 
     colA, colB = st.columns(2)
@@ -122,9 +122,8 @@ if uploaded_file is not None:
 
         plot_prediction(y_pred)
 
-        if download:
-            ccd = CCDData(y_pred, unit="adu", wcs=wcs)
-            ccd.write("predicted.fits", overwrite=True)
-            with open('predicted.fits', 'rb') as f:
-                data = f.read()
-            st.download_button(label="Download", data=data, file_name="predicted.fits", mime="application/octet-stream")
+        ccd = CCDData(y_pred, unit="adu", wcs=wcs)
+        ccd.write("predicted.fits", overwrite=True)
+        with open('predicted.fits', 'rb') as f:
+            data = f.read()
+        st.download_button(label="Download", data=data, file_name="predicted.fits", mime="application/octet-stream")