Update app.py

app.py

@@ -12,6 +12,7 @@ def initialize_auraflow_pipeline():
     ).to("cuda")
     return pipeline
 
+@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)