Spaces:

FrancescoLR
/

FLAMeS

Running on Zero

FrancescoLR commited on Dec 30, 2024

Commit

c32670e

1 Parent(s): 9ecae58

Updated app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -52,7 +52,7 @@ def run_nnunet_predict(nifti_file):
         "-d", "004",                  # Dataset ID
         "-c", "3d_fullres",           # Configuration
         "-tr", "nnUNetTrainer_8000epochs",
-        "--device", "cpu",
     ]
     try:
         subprocess.run(command, check=True)

         "-d", "004",                  # Dataset ID
         "-c", "3d_fullres",           # Configuration
         "-tr", "nnUNetTrainer_8000epochs",
+        "-device", "cpu",
     ]
     try:
         subprocess.run(command, check=True)