Update app.py

app.py

@@ -151,7 +151,7 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180, center=
     plt.close()
     
 # Function to run nnUNet inference
-@spaces.GPU(duration=70)  # Decorate the function to allocate GPU for its execution
+@spaces.GPU(duration=90)  # Decorate the function to allocate GPU for its execution
 def run_nnunet_predict(nifti_file):
     # Prepare directories
     os.makedirs(INPUT_DIR, exist_ok=True)
@@ -173,6 +173,7 @@ def run_nnunet_predict(nifti_file):
         "-i", input_path,
         "-o", hd_bet_output_path,
         "-device", "cuda",  # or "cpu"
+        "--disable_tta"   
         ], check=True)
        print("Skull-stripping completed.")
        input_path = hd_bet_output_path
@@ -194,7 +195,7 @@ def run_nnunet_predict(nifti_file):
         "-d", "004",                  # Dataset ID
         "-c", "3d_fullres",           # Configuration
         "-tr", "nnUNetTrainer_8000epochs",
-        "-device", "cuda"  # Explicitly use GPU
+        "-device", "cuda",  # Explicitly use GPU
     ]
     print("Files in /tmp/output:")
     print(os.listdir(OUTPUT_DIR))