Update app.py

app.py

@@ -155,6 +155,7 @@ def apply_seg(unet, child = None, blur_sigma=1.0):
             for sub_child in child.children():
                 apply_seg(unet, sub_child, blur_sigma=blur_sigma)
 
+@spaces.GPU
 def sample(prompt, cfg = 7.5, blur_sigma = blur_inf, seed=123, steps=50, signal_scale = 1.0):
 
   pipe = StableDiffusionPipeline.from_pretrained(model_id)