Update app.py

app.py

@@ -173,17 +173,24 @@ if uploaded_file is not None:
     #     st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
 
     if decompose:
-        y_pred = np.save("thresh.npy")
+        y_pred = np.load("thresh.npy")
         
         cavs = decompose_cavity(y_pred)
 
         ccd = CCDData(y_pred, unit="adu", wcs=wcs)
         ccd.write(f"predicted.fits", overwrite=True)
+        image_decomposed = np.zeros((128,128))
         for i, cav in enumerate(cavs):
             ccd = CCDData(cav, unit="adu", wcs=wcs)
             ccd.write(f"predicted_{i+1}.fits", overwrite=True)
 
-    plot_decomposed(np.zeros((128,128)))
+            image_decomposed += (i+1) * cav
+
+        np.save("decomposed.npy")
+
+    try: image_decomposed = np.load("decomposed.npy")
+    except: image_decomposed = np.zeros((128,128))
+    plot_decomposed(image_decomposed)
 
     #     with col4:
     #         pass