Update app.py

app.py

@@ -26,6 +26,7 @@ from huggingface_hub import hf_hub_download
 # model = model.cuda()
 # ddim_sampler = DDIMSampler(model)
 model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml')
+# ckpt = hf_hub_download(repo_id="xywwww/scene_diffusion", filename="checkpoints/epoch=25-step=112553.ckpt")
 ckpt = hf_hub_download(repo_id="xywwww/scene_diffusion", filename="checkpoints/epoch=29-step=479999.ckpt")
 model.load_state_dict(load_state_dict(ckpt), strict=False)
 model = model.cuda()