Update app.py

app.py

@@ -190,7 +190,7 @@ if uploaded_file is not None:
         
         cavs = decompose_cavity(y_pred)
 
-        ccd = CCDData(y_pred, unit="adu", wcs=wcs)
+        # ccd = CCDData(y_pred, unit="adu", wcs=wcs)
         # ccd.write(f"predictions/predicted.fits", overwrite=True)
         image_decomposed = np.zeros((128,128))
         for i, cav in enumerate(cavs):
@@ -210,6 +210,7 @@ if uploaded_file is not None:
     # shutil.make_archive("predictions", 'zip', "predictions")
 
     with col6:
+        ccd = CCDData(y_pred, unit="adu", wcs=wcs)
         # with open('predictions.zip', 'rb') as f:
         #     res = f.read()
     #     # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)