Upload folder using huggingface_hub

jolia_atlas_transform.py

@@ -10,7 +10,10 @@ from typing import TYPE_CHECKING, Sequence, Union
 
 import torch
 import torch.nn.functional as F
-from .jolia_windowing import batch_apply_windowing_vectorized
+try:
+    from .jolia_windowing import batch_apply_windowing_vectorized
+except ImportError:
+    from jolia_windowing import batch_apply_windowing_vectorized
 
 if TYPE_CHECKING:
     import numpy as np

preprocessing_jolia.py

@@ -21,14 +21,26 @@ from typing import Union
 
 import torch
 
-from .jolia_atlas_transform import (
-    ApplyWindowing,
-    AtlasTransform,
-    Crop3D,
-    Pad3D,
-    PrepareVolume,
-    Resample3D,
-)
+# Works both inside a package (HF trust_remote_code) and as a top-level module
+# (the `snapshot_download` + `sys.path.append` flow in the README).
+try:
+    from .jolia_atlas_transform import (
+        ApplyWindowing,
+        AtlasTransform,
+        Crop3D,
+        Pad3D,
+        PrepareVolume,
+        Resample3D,
+    )
+except ImportError:
+    from jolia_atlas_transform import (
+        ApplyWindowing,
+        AtlasTransform,
+        Crop3D,
+        Pad3D,
+        PrepareVolume,
+        Resample3D,
+    )
 
 try:  # numpy is optional at import time; only needed for ndarray inputs
     import numpy as np