Update app.py

app.py

@@ -3,6 +3,7 @@ from diffusers import AuraFlowPipeline
 import torch
 import gradio as gr
 
+@spaces.GPU()
 def initialize_auraflow_pipeline():
     """Initialize and return the AuraFlowPipeline."""
     pipeline = AuraFlowPipeline.from_pretrained(
@@ -12,7 +13,7 @@ def initialize_auraflow_pipeline():
     ).to("cuda")
     return pipeline
 
-@spaces.GPU(duration=100)
+@spaces.GPU()
 def generate_image(pipeline, prompt, width, height, num_inference_steps, seed, guidance_scale):
     """Generate an image using the AuraFlowPipeline."""
     generator = torch.Generator().manual_seed(seed)