Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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", "
|
| 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
|