diff --git a/monai/apps/mmars/model_desc.py b/monai/apps/mmars/model_desc.py index 97846b067c..2142788f80 100644 --- a/monai/apps/mmars/model_desc.py +++ b/monai/apps/mmars/model_desc.py @@ -184,4 +184,26 @@ class RemoteMMARKeys: RemoteMMARKeys.HASH_VAL: None, RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"), }, + { + RemoteMMARKeys.ID: "clara_pt_deepgrow_3d_annotation_1", + RemoteMMARKeys.NAME: "clara_pt_deepgrow_3d_annotation", + RemoteMMARKeys.URL: "https://api.ngc.nvidia.com/v2/models/nvidia/med/clara_pt_deepgrow_3d_annotation/versions/1/zip", + RemoteMMARKeys.DOC: "https://ngc.nvidia.com/catalog/models/nvidia:med:clara_pt_deepgrow_3d_annotation", + RemoteMMARKeys.FILE_TYPE: "zip", + RemoteMMARKeys.HASH_TYPE: "md5", + RemoteMMARKeys.HASH_VAL: None, + RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"), + RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"), + }, + { + RemoteMMARKeys.ID: "clara_pt_deepgrow_2d_annotation_1", + RemoteMMARKeys.NAME: "clara_pt_deepgrow_2d_annotation", + RemoteMMARKeys.URL: "https://api.ngc.nvidia.com/v2/models/nvidia/med/clara_pt_deepgrow_2d_annotation/versions/1/zip", + RemoteMMARKeys.DOC: "https://ngc.nvidia.com/catalog/models/nvidia:med:clara_pt_deepgrow_2d_annotation", + RemoteMMARKeys.FILE_TYPE: "zip", + RemoteMMARKeys.HASH_TYPE: "md5", + RemoteMMARKeys.HASH_VAL: None, + RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"), + RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"), + }, )