FrancescoLR commited on
Commit
94ba9ed
·
1 Parent(s): 5dd0ea6

Updated app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- "--device", "cuda:0" # Explicitly use GPU 0
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