Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from diffusers import StableDiffusionXLPipeline, UNet2DConditionModel
 import torch
 
@@ -15,6 +16,7 @@ pipe = pipe.to("cuda")
 pipe.enable_model_cpu_offload() 
 pipe.enable_vae_slicing()
 
+@spaces.GPU
 def infer(prompt):
     image = pipe(prompt, guidance_scale=7.5).images[0]
     return image