Update app.py

app.py

@@ -227,10 +227,12 @@ if uploaded_file is not None:
             except: image_decomposed = np.zeros((128,128))
             plot_decomposed(image_decomposed)
         
-    with col6:
-        shutil.make_archive("predictions", 'zip', "predictions")
-        with open('predictions.zip', 'rb') as f:
-            res = f.read()
-        st.markdown("")
-        # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
-        download = st.download_button(label="Download", data=res, file_name='prediction.zip', mime="application/octet-stream")
+        with col6:
+            shutil.make_archive("predictions", 'zip', "predictions")
+            with open('predictions.zip', 'rb') as f:
+                res = f.read()
+            st.markdown("")
+            st.markdown("")
+            st.markdown("")
+            # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
+            download = st.download_button(label="Download", data=res, file_name='prediction.zip', mime="application/octet-stream")