FrancescoLR commited on
Commit
3ebeec1
·
verified ·
1 Parent(s): c0af38a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -172,8 +172,7 @@ def run_nnunet_predict(nifti_file):
172
  "hd-bet",
173
  "-i", input_path,
174
  "-o", hd_bet_output_path,
175
- "-device", "cpu", # or "cuda" if available
176
- "-mode", "fast"
177
  ], check=True)
178
  print("Skull-stripping completed.")
179
  input_path = hd_bet_output_path
 
172
  "hd-bet",
173
  "-i", input_path,
174
  "-o", hd_bet_output_path,
175
+ "-device", "cuda", # or "cpu"
 
176
  ], check=True)
177
  print("Skull-stripping completed.")
178
  input_path = hd_bet_output_path