Update app.py

app.py

@@ -28,6 +28,7 @@ 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
@@ -37,6 +38,7 @@ 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")