Updated app.py

app.py

@@ -35,9 +35,7 @@ def run_nnunet_predict(nifti_file):
     os.rename(nifti_file.name, input_path)  # Move the uploaded file to the expected input location
 
     # Set environment variables for nnUNet
-    os.environ["nnUNet_raw"] = DATASET_DIR
-    os.environ["nnUNet_preprocessed"] = DATASET_DIR
-    os.environ["nnUNet_results"] = DATASET_DIR
+    os.environ["nnUNet_results"] = MODEL_DIR
 
     # Construct and run the nnUNetv2_predict command
     command = [