Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -175,10 +175,10 @@ def run_nnunet_predict(nifti_file):
|
|
| 175 |
"-device", "cpu", # or "cuda" if available
|
| 176 |
"-mode", "fast"
|
| 177 |
], check=True)
|
| 178 |
-
|
| 179 |
-
|
| 180 |
except subprocess.CalledProcessError as e:
|
| 181 |
-
|
| 182 |
|
| 183 |
# Debugging: List files in the /tmp/input directory
|
| 184 |
print("Files in /tmp/input:")
|
|
|
|
| 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
|
| 180 |
except subprocess.CalledProcessError as e:
|
| 181 |
+
return f"HD-BET Error: {e}"
|
| 182 |
|
| 183 |
# Debugging: List files in the /tmp/input directory
|
| 184 |
print("Files in /tmp/input:")
|