Update app.py

app.py

@@ -84,7 +84,7 @@ def cut(data0, wcs0, scale=1):
     return data, wcs
 
 # Define function to apply cutting and produce a prediction
-@st.cache_data
+@st.cache #_data
 def cut_n_predict(data, _wcs, scale):
     data, wcs = cut(data, _wcs, scale=scale)
     image = np.log10(data+1)
@@ -99,7 +99,7 @@ def cut_n_predict(data, _wcs, scale):
     return y_pred, wcs
 
 # Define function to decompose prediction into individual cavities
-@st.cache_data
+@st.cache #_data
 def decompose_cavity(pred, fname, th2=0.7, amin=10):
     X, Y = pred.nonzero()
     data = np.array([X,Y]).reshape(2, -1)
@@ -138,7 +138,7 @@ def decompose_cavity(pred, fname, th2=0.7, amin=10):
     
     return image_decomposed
 
-@st.cache_data
+@st.cache #_data
 def load_file(fname):
     with fits.open(fname) as hdul:
         data = hdul[0].data