Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 import numpy as np
 import random
 from diffusers import AutoPipelineForText2Image
@@ -24,6 +25,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, progress=gr.Progress(track_tqdm=True)):
 
     if randomize_seed: