Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import random
 from diffusers import AuraFlowPipeline
 import torch
+import spaces
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -19,6 +20,7 @@ else:
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
+@spaces.GPU
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
 
     if randomize_seed: