Update app.py

app.py

@@ -135,12 +135,12 @@ def decompose_cavity(pred, th2=0.7, amin=6):
     
     return image_decomposed
 
-# @st.cache
-# def load_file(fname):
-#     with fits.open(fname) as hdul:
-#         data = hdul[0].data
-#         wcs = WCS(hdul[0].header)
-#     return data, wcs
+@st.cache
+def load_file(fname):
+    with fits.open(fname) as hdul:
+        data = hdul[0].data
+        wcs = WCS(hdul[0].header)
+    return data, wcs
 
 def change_scale():
     del st.session_state["threshold"]
@@ -188,10 +188,10 @@ with col:
 #     data = np.zeros((128,128))
 
 if uploaded_file is not None:
-    # data, wcs = load_file(uploaded_file)
-    with fits.open(uploaded_file) as hdul:
-        data = hdul[0].data
-        wcs = WCS(hdul[0].header)
+    data, wcs = load_file(uploaded_file)
+    # with fits.open(uploaded_file) as hdul:
+    #     data = hdul[0].data
+    #     wcs = WCS(hdul[0].header)
     
     # Make six columns for buttons
     _, col1, col2, col3, col4, col5, col6, _ = st.columns([bordersize,0.5,0.5,0.5,0.5,0.5,0.5,bordersize])