feat: Update link to download new model

Kohulan · Nov 16, 2023 · 28c3397 · 28c3397
1 parent e648541
commit 28c3397
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/decimer_segmentation/decimer_segmentation.py b/decimer_segmentation/decimer_segmentation.py
@@ -204,7 +204,7 @@ def load_model() -> modellib.MaskRCNN:
     # Download trained weights if needed
     if not os.path.exists(model_path):
         print("Downloading model weights...")
-        url = "https://zenodo.org/record/7228583/files/mask_rcnn_molecule.h5?download=1"
+        url = "https://zenodo.org/records/10142866/files/mask_rcnn_molecule.h5?download=1"
         req = requests.get(url, allow_redirects=True)
         with open(model_path, "wb") as model_file:
             model_file.write(req.content)