Spaces:
Running on Zero
Running on Zero
Francesco commited on
Commit ·
94ba9ed
1
Parent(s): 5dd0ea6
Updated app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ def run_nnunet_predict(nifti_file):
|
|
| 50 |
"-d", "004", # Dataset ID
|
| 51 |
"-c", "3d_fullres", # Configuration
|
| 52 |
"-tr", "nnUNetTrainer_8000epochs",
|
| 53 |
-
"
|
| 54 |
]
|
| 55 |
try:
|
| 56 |
subprocess.run(command, check=True)
|
|
@@ -84,5 +84,5 @@ download_model()
|
|
| 84 |
|
| 85 |
# Launch the app
|
| 86 |
if __name__ == "__main__":
|
| 87 |
-
interface.launch()
|
| 88 |
|
|
|
|
| 50 |
"-d", "004", # Dataset ID
|
| 51 |
"-c", "3d_fullres", # Configuration
|
| 52 |
"-tr", "nnUNetTrainer_8000epochs",
|
| 53 |
+
"-device", "cuda:0" # Explicitly use GPU 0
|
| 54 |
]
|
| 55 |
try:
|
| 56 |
subprocess.run(command, check=True)
|
|
|
|
| 84 |
|
| 85 |
# Launch the app
|
| 86 |
if __name__ == "__main__":
|
| 87 |
+
interface.launch(share=True)
|
| 88 |
|