Fix Bug

app.py

@@ -26,9 +26,10 @@ model_cfg = get_config(path_model_cfg)
 # Load model
 repo_id = "luoxue-star/AniMer"
 local_dir = snapshot_download(repo_id=repo_id)
+# local_dir = "./checkpoints"
 PATH_CHECKPOINT = os.path.join(local_dir, "checkpoint.ckpt")
 model = AMR.load_from_checkpoint(checkpoint_path=PATH_CHECKPOINT, map_location="cpu",
-                                 cfg=model_cfg, strict=False)
+                                 cfg=model_cfg, strict=False, weights_only=True)
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 model = model.to(device)
 model.eval()