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])