revert change in tcia

Signed-off-by: YunLiu <[email protected]>

Author: KumoLiu

model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb

@@ -500,7 +500,7 @@
    },
    "outputs": [],
    "source": [
-    "model.load_state_dict(torch.load(checkpoint, map_location=device, weights_only=True))\n",
+    "model.load_state_dict(torch.load(checkpoint, map_location=device))\n",
     "model.eval()\n",
     "results = []\n",
     "with torch.no_grad():\n",