Update app.py

app.py

@@ -21,7 +21,7 @@ import numpy as np
 import cv2
 import os
 import random
-
+import spaces
 
 def nms(x, t, s):
     x = cv2.GaussianBlur(x.astype(np.float32), (0, 0), s)
@@ -158,7 +158,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
-
+@spaces.GPU
 def run(
     image: PIL.Image.Image,
     prompt: str,