added else statement

.github/workflows/build.yml

@@ -56,7 +56,7 @@ jobs:
       - name: Install software
         run: pip install --find-links=${{github.workspace}} lungtumormask
 
-      - name: Download test data
+      - name: Download test data 
         run: |
           pip install gdown
           mkdir samples
@@ -69,4 +69,3 @@ jobs:
 
       - name: Test inference
         run: lungtumormask samples/lung_001.nii.gz mask_001.nii.gz
-        

lungtumormask/dataprocessing.py

@@ -12,6 +12,8 @@ def mask_lung(scan_path, batch_size=20):
     model = lungmask.mask.get_model('unet', 'R231')
     if torch.cuda.is_available():
         device = torch.device('cuda')
+    else:
+        device = torch.device('cpu')
     model.to(device)
 
     scan_dict = {