Update app.py

app.py

@@ -127,7 +127,7 @@ def process_nii_file(input_nii_file, slice, mode):
 
         with torch.no_grad():
 
-            inputs = torch.unsqueeze(test_image, 1).cuda()
+            inputs = torch.unsqueeze(test_image, 1).cpu() ################################### .cuda() -> .cpu()
 
             val_outputs = sliding_window_inference(inputs, (96, 96, 16), 8, model, overlap=0.8)