Update app.py

app.py

@@ -9,8 +9,9 @@ from huggingface_hub import hf_hub_download
 # config = "./models/cldm_v21_512_latctrl_coltrans.yaml'"
 model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml').cpu()
 ckpt = hf_hub_download(repo_id="xywwww/scene_diffusion", filename="checkpoints/epoch=25-step=112553.ckpt")
-# model.load_state_dict(load_state_dict('xywwww/scene_diffusion/checkpoints/epoch=25-step=112553.ckpt', location='cuda'), strict=False)
-model = load_model_checkpoint(model, ckpt)
+print(ckpt)
+model.load_state_dict(load_state_dict(ckpt, location='cuda'), strict=False)
+# model = load_model_checkpoint(model, ckpt)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)