fix: 🐛 Fix model download and loading paths.

src/cellmap_models/pytorch/cellpose/get_model.py

@@ -20,11 +20,9 @@ def get_model(
         raise ValueError(
             f"Model {model_name} is not available. Available models are {list(models_dict.keys())}."
         )
-
-    if not Path(base_path / f"{model_name}.pth").exists():
+    full_path = os.path.join(base_path, f"{model_name}.pth")
+    if not Path(full_path).exists():
         print(f"Downloading {model_name} from {models_dict[model_name]}")
-        download_url_to_file(
-            models_dict[model_name], os.path.join(base_path, f"{model_name}.pth")
-        )
+        download_url_to_file(models_dict[model_name], full_path)
     print("Downloaded model {model_name} to {base_path}.")
     return

src/cellmap_models/pytorch/cellpose/load_model.py

@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 import torch
 from .get_model import get_model
@@ -23,6 +24,6 @@ def load_model(
     if device == "cuda" and not torch.cuda.is_available():
         device = "cpu"
         print("CUDA not available. Using CPU.")
-    model = torch.jit.load(str(base_path / f"{model_name}.pth"), device)
+    model = torch.jit.load(os.path.join(base_path, f"{model_name}.pth"), device)
     model.eval()
     return model