feat: add spaces import for zero gpu

app.py

@@ -1,4 +1,5 @@
 import torch
+import spaces
 from diffusers import StableDiffusionPipeline
 import gradio as gr
 
@@ -6,6 +7,9 @@ repo = "IDKiro/sdxs-512-0.9"
 seed = 42
 weight_type = torch.float16
 
+zero = torch.Tensor([0]).cuda()
+print(zero.device) # <-- 'cpu' 🤔
+
 # Load model.
 pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
 
@@ -15,7 +19,7 @@ generator = pipe
 if torch.cuda.is_available():
     generator = generator.to("cuda")
 
-
+@spaces.GPU(duration=120)
 def generate(prompts):
     images = generator(list(prompts)).images
     return [images]