update app.py

app.py

@@ -52,24 +52,31 @@ def add_text(rgb, caption):
     return rgb
 
 config = "3DTopia/configs/default.yaml"
-local_ckpt = "checkpoints/3dtopia_diffusion_state_dict.ckpt"
+local_ckpt = "3DTopia/checkpoints/3dtopia_diffusion_state_dict.ckpt"
 if os.path.exists(local_ckpt):
     ckpt = local_ckpt
 else:
     ckpt = hf_hub_download(repo_id="hongfz16/3DTopia", filename="model.safetensors")
+print("download finish")
 configs = OmegaConf.load(config)
 os.makedirs("tmp", exist_ok=True)
+print("download finish")
 
 if ckpt.endswith(".ckpt"):
     model = get_obj_from_str(configs.model["target"]).load_from_checkpoint(ckpt, map_location='cpu', strict=False, **configs.model.params)
 elif ckpt.endswith(".safetensors"):
+    print("download finish")
     model = get_obj_from_str(configs.model["target"])(**configs.model.params)
+    print("download finish")
     model_ckpt = load_file(ckpt)
+    print("download finish")
     model.load_state_dict(model_ckpt)
+    print("download finish")
 else:
     raise NotImplementedError
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 model = model.to(device)
+print("download finish")
 sampler = DDIMSampler(model)
 
 img_size = configs.model.params.unet_config.params.image_size
@@ -106,6 +113,7 @@ for p in poses_fname:
     batch_rays_list.append(batch_rays)
 batch_rays_list = torch.stack(batch_rays_list, 0)
 
+print("download finish")
 def marching_cube(b, text, global_info):
     # prepare volumn for marching cube
     res = 128