Spaces:

FrancescoLR
/

FLAMeS

Running on Zero

FrancescoLR commited on Dec 30, 2024

Commit

2bce9bd

1 Parent(s): 94ba9ed

Updated app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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