initialize controlnet with decorator

app.py

@@ -10,11 +10,9 @@ from game_of_life import GameOfLife
 from utils import resize_image, generate_image_from_grid
 
 
-device = "cuda"
-#  device = "mps"
-#  device = "cpu"
-print(f"Using {device=}")
-controlnet = QRControlNet(device=device)
+@spaces.gpu
+def init_controlnet(device: str) -> QRControlNet:
+    return QRControlNet(device=device)
 
 
 def generate_all_images(
@@ -113,6 +111,13 @@ def generate(
     return path_gol_controlnet, path_gol_gif
 
 
+device = "cuda"
+#  device = "mps"
+#  device = "cpu"
+print(f"Using {device=}")
+controlnet = init_controlnet()
+
+
 source_image = gr.Image(label="Source Image", type="pil", value="sky-gol-image.jpeg")
 
 output_controlnet = gr.Image(label="ControlNet Game of Life")