Updated app.py

app.py

@@ -31,8 +31,12 @@ 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.nii.gz")
+    input_path = os.path.join(INPUT_DIR, "image_0000.nii.gz")
     os.rename(nifti_file.name, input_path)  # Move the uploaded file to the expected input location
+    
+    # Debugging: List files in the /tmp/input directory
+    print("Files in /tmp/input:")
+    print(os.listdir(INPUT_DIR))
 
     # Set environment variables for nnUNet
     os.environ["nnUNet_results"] = MODEL_DIR