FrancescoLR commited on
Commit
81ef1c7
·
verified ·
1 Parent(s): ee82fb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import torch.nn as nn
8
  from huggingface_hub import hf_hub_download
9
  from monai.transforms import Compose, LoadImaged, Spacingd, CropForegroundd, SpatialPadd, CenterSpatialCropd
10
  from monai.data import Dataset
11
- from nnunet_mednext import create_mednext_encoder_v1
12
 
13
  # Device selection
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
8
  from huggingface_hub import hf_hub_download
9
  from monai.transforms import Compose, LoadImaged, Spacingd, CropForegroundd, SpatialPadd, CenterSpatialCropd
10
  from monai.data import Dataset
11
+ from mednextv1 import create_mednext_encoder_v1
12
 
13
  # Device selection
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")