Update app.py

app.py

@@ -12,6 +12,9 @@ from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 # Create a Flask instance
 app = Flask(__name__)
 
+def clear_gpu_memory():
+    torch.cuda.empty_cache()
+    torch.cuda.synchronize()
 
 # Initialize model only once
 pipe = None
@@ -101,7 +104,7 @@ def run():
     num_images_per_prompt = data['num_images_per_prompt']
     width = data['width'] if 'width' in data else 512  # Default width
     height = data['height'] if 'height' in data else 512  # Default height
-    
+    clear_gpu_memory()
     result = generate(
         prompt,
         negative_prompt,