Update app.py

app.py

@@ -11,7 +11,7 @@ from abc import abstractmethod
 import torch.backends
 import torch.mps
 from PIL import Image
-
+import spaces
 
 if torch.cuda.is_available():
     device = torch.device("cuda")
@@ -185,7 +185,7 @@ def describe_shape(shape):
     return f"Shape: {shape} num elements: {np.prod(shape)}"
 
 
-# @spaces.GPU
+@spaces.GPU(duration=32)
 @torch.no_grad()
 def roundtrip_image(
     image,