Update app.py

app.py

@@ -22,7 +22,8 @@ pose_checkpoint = "models/control_sd15_openpose.pth"
 canny_model = create_model('./models/cldm_v15.yaml')
 canny_model.load_state_dict(load_state_dict(cached_download(
     hf_hub_url(REPO_ID, canny_checkpoint)
-), location='cuda'))
+), location='cpu'))
+canny_model = canny_model.cuda()
 ddim_sampler = DDIMSampler(canny_model)