Updated download path to point to the trained_models_v1 release

DeepDeformationMapRegistration/utils/model_utils.py

@@ -37,7 +37,7 @@ def get_models_path(anatomy: str, model_type: str, output_root_dir: str):
     assert model_type in MODEL_TYPES.keys(), 'Invalid model type'
     anatomy = ANATOMIES[anatomy]
     model_type = MODEL_TYPES[model_type]
-    url = 'https://github.com/jpdefrutos/DDMR/releases/download/trained_models_v0/' + anatomy + '_' + model_type + '.h5'
+    url = 'https://github.com/jpdefrutos/DDMR/releases/download/trained_models_v1/' + anatomy + '_' + model_type + '.h5'
     file_path = os.path.join(output_root_dir, 'models', anatomy, model_type + '.h5')
     if not os.path.exists(file_path):
         LOGGER.info(f'Model not found. Downloading from {url}... ')
@@ -66,7 +66,7 @@ def load_model(weights_file_path: str, trainable: bool = False, return_registrat
 
 
 def get_spatialtransformer_model():
-    url = 'https://github.com/jpdefrutos/DDMR/releases/download/trained_models_v0/spatialtransformer.h5'
+    url = 'https://github.com/jpdefrutos/DDMR/releases/download/trained_models_v1/spatialtransformer.h5'
     file_path = os.path.join(os.getcwd(), 'models', 'spatialtransformer.h5')
     if not os.path.exists(file_path):
         LOGGER.info(f'Model not found. Downloading from {url}... ')