FrancescoLR commited on
Commit
ed9fa70
·
1 Parent(s): 4d847b9

Updated app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -35,9 +35,7 @@ def run_nnunet_predict(nifti_file):
35
  os.rename(nifti_file.name, input_path) # Move the uploaded file to the expected input location
36
 
37
  # Set environment variables for nnUNet
38
- os.environ["nnUNet_raw"] = DATASET_DIR
39
- os.environ["nnUNet_preprocessed"] = DATASET_DIR
40
- os.environ["nnUNet_results"] = DATASET_DIR
41
 
42
  # Construct and run the nnUNetv2_predict command
43
  command = [
 
35
  os.rename(nifti_file.name, input_path) # Move the uploaded file to the expected input location
36
 
37
  # Set environment variables for nnUNet
38
+ os.environ["nnUNet_results"] = MODEL_DIR
 
 
39
 
40
  # Construct and run the nnUNetv2_predict command
41
  command = [