Update app.py

app.py

@@ -18,6 +18,7 @@ from PIL import Image, ImageOps, ImageFilter
 from torch import autocast
 import cv2
 import imageio
+import spaces
 
 sys.path.append("./stable_diffusion")
 
@@ -155,6 +156,7 @@ model_wrap = CompVisDenoiser(model)
 model_wrap_cfg = CFGDenoiser(model_wrap)
 null_token = model.get_learned_conditioning([""])
 
+@spaces.GPU(duration=20)
 def generate(
     input_image: Image.Image,
     instruction: str,