fix: decimer-segmentation improved model

Kohulan · Feb 15, 2024 · 7bd9601 · 7bd9601
1 parent ca05e92
commit 7bd9601
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/decimer_segmentation/decimer_segmentation.py b/decimer_segmentation/decimer_segmentation.py
@@ -173,7 +173,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/record/10663579/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)