ZeroGPU changes

app.py

@@ -27,12 +27,10 @@ from src.pipelines.pipeline_kandinsky_subject_prior import KandinskyPriorPipelin
 from diffusers import DiffusionPipeline
 from PIL import Image
 import random
+import spaces
 
-__device__ = "cpu"
-__dtype__ = torch.float32
-if torch.cuda.is_available():
-    __device__ = "cuda"
-    __dtype__ = torch.float16
+__device__ = "cuda"
+__dtype__ = torch.float16
 
 
 class Model:
@@ -85,6 +83,7 @@ class Model:
         ).images[0]
         return image
 
+    @spaces.GPU
     def run(
         self,
         image: dict[str, PIL.Image.Image],