minor update to fxn call

Author: MMathisLab

napari_cellseg3d/models/TRAILMAP_MS.py

@@ -7,7 +7,7 @@
 target_dir = os.path.join("models","pretrained")
 
 def get_weights_file():
-    utils.DownloadModel(model, train_dir)
+    utils.DownloadModel(modelname, target_dir)
     return "TRAILMAP_MS_best_metric_epoch_26.pth" #model additionally trained on Mathis/Wyss mesoSPIM data
 
 

napari_cellseg3d/models/model_TRAILMAP.py

@@ -6,7 +6,7 @@
 target_dir = os.path.join("models","pretrained")
 
 def get_weights_file():
-    utils.DownloadModel(model, train_dir)
+    uutils.DownloadModel(modelname, target_dir)
     return "TRAILMAP_PyTorch.pth" #original model form Luo lab, transfered to pytroch
 
 

napari_cellseg3d/models/model_VNet.py

@@ -11,7 +11,7 @@ def get_net():
 
 
 def get_weights_file():
-    utils.DownloadModel(model, train_dir)
+    utils.DownloadModel(modelname, target_dir)
     return "VNet_40e.pth"