Update app.py

app.py

@@ -189,6 +189,9 @@ if uploaded_file is not None:
         shutil.make_archive("predictions", 'zip', "predictions")
         with open('predictions.zip', 'rb') as f:
             res = f.read()
+
+       if st.session_state.get("decompose", False):
+            st.session_state.disabled = False
         download = st.download_button(label="Download", data=res, 
                                       file_name=f'{fname}_{int(scale*128)}.zip', 
                                       disabled=st.session_state.get("disabled", True), 
@@ -211,7 +214,4 @@ if uploaded_file is not None:
         if decompose or download:            
             image_decomposed = decompose_cavity(y_pred_th)
     
-            plot_decomposed(image_decomposed)
-
-        if st.session_state.get("decompose", True):
-            st.session_state.disabled = False
+            plot_decomposed(image_decomposed)