Update app.py

app.py

@@ -6,6 +6,8 @@ import torch
 pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
 pipe.to("cpu")
 
+# Cache the image generation function
+@st.cache(suppress_st_warning=True)
 def generate_image(prompt):
     # Generate image using the model
     image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
@@ -23,3 +25,4 @@ if st.button("Generate Image"):
         st.image(generate_image(prompt), caption="Generated Image", use_column_width=True)
     else:
         st.warning("Please enter a prompt.")
+