Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 import random
 from diffusers import AutoencoderKL, DiffusionPipeline
 import torch
-
+import spaces
 import os
 import PIL.Image 
 
@@ -56,7 +56,7 @@ if torch.cuda.is_available():
         if ENABLE_REFINER:
             refiner.unet = torch.compile(refiner.unet, mode="reduce_overhead", fullgraph=True)
 
-
+@spaces.GPU(enable_queue=True)
 def infer(
     prompt: str,
     seed: int = 0,