Update app.py

app.py

@@ -28,7 +28,6 @@ st.set_page_config(page_title="Cavity Detection Tool", layout="wide")
 # st.title("Cavity Detection Tool")
 
 # Define function to plot the uploaded image
-@st.cache
 def plot_image(image, scale):
     plt.figure(figsize=(4, 4))
     x0 = image.shape[0] // 2 - scale * 128 / 2
@@ -38,7 +37,6 @@ def plot_image(image, scale):
     with colA: st.pyplot()
 
 # Define function to plot the prediction
-@st.cache
 def plot_prediction(pred):
     plt.figure(figsize=(4, 4))
     plt.imshow(pred, origin="lower")
@@ -99,6 +97,7 @@ def cut_n_predict(data, wcs, scale):
     return y_pred, wcs
 
 # Define function to decomposed prediction into cavities
+@st.cache
 def decompose_cavity(pred, th2=0.7, amin=10):
     X, Y = pred.nonzero()
     data = np.array([X,Y]).reshape(2, -1)