Update app.py

app.py

@@ -78,6 +78,7 @@ Below is the simple caption of an image with text. Please deduce the detailed de
 @spaces.GPU(duration=60)
 def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
     pipe.to("cuda")
+    pipe.enable_model_cpu_offload()
     """Generate image using LeX-Lumina"""
     # Truncate the caption if it's too long
     enhanced_caption = truncate_caption_by_tokens(enhanced_caption, max_tokens=256)