Updated app.py

app.py

@@ -31,12 +31,14 @@ def run_nnunet_predict(nifti_file):
     os.makedirs(OUTPUT_DIR, exist_ok=True)
 
     # Save the uploaded file to the input directory
-    input_path = os.path.join(INPUT_DIR, "image_0000.nii.gz")
+    input_path = os.path.join(INPUT_DIR, "image.nii.gz")
     with open(input_path, "wb") as f:
-        f.write(nifti_file.read())
+        f.write(nifti_file.file.read())  # Use .file to access the underlying file object
 
     # Set environment variables for nnUNet
-    os.environ["nnUNet_results"] = MODEL_DIR
+    os.environ["nnUNet_raw"] = DATASET_DIR
+    os.environ["nnUNet_preprocessed"] = DATASET_DIR
+    os.environ["nnUNet_results"] = DATASET_DIR
 
     # Construct and run the nnUNetv2_predict command
     command = [