Update app.py

app.py

@@ -9,6 +9,7 @@ import torch
 from PIL import Image, ImageOps
 import time
 import psutil
+import spaces
 import math
 import random
 
@@ -20,7 +21,7 @@ device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
 if torch.cuda.is_available():
     pipe = pipe.to("cuda")
 
-
+@spaces.GPU(duration=120)
 def pix2pix(
         input_image: Image.Image,
         instruction: str,