Add pipe.to("cuda")

app.py

@@ -37,7 +37,9 @@ def predict(image_input, prompt, negative_prompt, sd_models_dropdown):
         controlnet=controlnet,
         torch_dtype=torch.float16
     )
-
+    
+    pipe = pipe.to("cuda")
+    
     # speed up diffusion process with faster scheduler and memory optimization
     pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)