Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from diffusers import DiffusionPipeline
 
 # Load the Diffusion pipeline
-@st.cache(allow_output_mutation=True)
+@st.cache_data(allow_output_mutation=True)
 def load_diffusion_pipeline():
     try:
         pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium")
@@ -38,4 +38,3 @@ def main():
 
 if __name__ == '__main__':
     main()
-