FrancescoLR commited on
Commit
e133e90
·
verified ·
1 Parent(s): 73bc6a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import nibabel as nib
6
  import torchio
7
  import torch.nn as nn
8
  import subprocess
 
9
  from scipy.ndimage.measurements import center_of_mass
10
  from huggingface_hub import hf_hub_download
11
  from monai.transforms import Compose, LoadImaged, Spacingd, CropForegroundd, SpatialPadd, CenterSpatialCropd
 
6
  import torchio
7
  import torch.nn as nn
8
  import subprocess
9
+ import spaces # Import spaces for GPU decoration
10
  from scipy.ndimage.measurements import center_of_mass
11
  from huggingface_hub import hf_hub_download
12
  from monai.transforms import Compose, LoadImaged, Spacingd, CropForegroundd, SpatialPadd, CenterSpatialCropd