fix GPU decorator, add missing arg

app.py

@@ -10,7 +10,7 @@ from game_of_life import GameOfLife
 from utils import resize_image, generate_image_from_grid
 
 
-@spaces.gpu
+@spaces.GPU
 def init_controlnet(device: str) -> QRControlNet:
     return QRControlNet(device=device)
 
@@ -112,10 +112,10 @@ def generate(
 
 
 device = "cuda"
-#  device = "mps"
-#  device = "cpu"
+# device = "mps"
+# device = "cpu"
 print(f"Using {device=}")
-controlnet = init_controlnet()
+controlnet = init_controlnet(device=device)
 
 
 source_image = gr.Image(label="Source Image", type="pil", value="sky-gol-image.jpeg")