Spaces:
Sleeping
Sleeping
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 |
|