Update model_SwinUNetR.py

Author: C-Achard

napari_cellseg3d/code_models/models/model_SwinUNetR.py

@@ -32,7 +32,7 @@ def __init__(
         """
         try:
             super().__init__(
-                # img_size=input_img_size,
+                img_size=input_img_size,
                 in_channels=in_channels,
                 out_channels=out_channels,
                 feature_size=48,
@@ -45,7 +45,6 @@ def __init__(
         except TypeError as e:
             logger.warning(f"Caught TypeError: {e}")
             super().__init__(
-                img_size=input_img_size, # hack for compatibility, latest MONAI for py3.9 does not have img_size but py3.8 does
                 in_channels=1,
                 out_channels=1,
                 feature_size=48,