Update app.py

app.py

@@ -32,7 +32,8 @@ styles = {
 }
 
 # Cache the pipeline to avoid reloading it multiple times
-@gr.memo
+from functools import lru_cache
+@lru_cache(maxsize=1)
 def load_pipeline(use_lora: bool):
     """Load the diffusion pipeline with or without LoRA weights."""
     device = "cuda" if torch.cuda.is_available() else "cpu"