Update app.py

app.py

@@ -125,12 +125,12 @@ def decompose_cavity(pred, th2=0.7, amin=10):
 
     return cavities
 
-@st.cache
-def zip_predictions():
-    shutil.make_archive("predictions.zip", 'zip', "predictions")
-    with open('predictions.zip', 'rb') as f:
-        res = f.read()
-        return res
+# @st.cache
+# def zip_predictions():
+#     shutil.make_archive("predictions.zip", 'zip', "predictions")
+#     with open('predictions.zip', 'rb') as f:
+#         res = f.read()
+#         return res
 
 bordersize = 0.6
 _, col, _ = st.columns([bordersize, 3, bordersize])
@@ -219,6 +219,8 @@ if uploaded_file is not None:
         plot_decomposed(image_decomposed)
     
     with col6:
-        res = zip_predictions()
+        shutil.make_archive("predictions.zip", 'zip', "predictions")
+        with open('predictions.zip', 'rb') as f:
+            res = f.read()
         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.fits', mime="application/octet-stream")