Updated app.py

app.py

@@ -37,6 +37,9 @@ def run_nnunet_predict(nifti_file):
     # Debugging: List files in the /tmp/input directory
     print("Files in /tmp/input:")
     print(os.listdir(INPUT_DIR))
+    
+    print("Files in /tmp/output:")
+    print(os.listdir(OUTPUT_DIR))
 
     # Set environment variables for nnUNet
     os.environ["nnUNet_results"] = MODEL_DIR
@@ -48,7 +51,8 @@ def run_nnunet_predict(nifti_file):
         "-o", OUTPUT_DIR,
         "-d", "004",                  # Dataset ID
         "-c", "3d_fullres",           # Configuration
-        "-tr", "nnUNetTrainer_8000epochs",  # Trainer name
+        "-tr", "nnUNetTrainer_8000epochs",
+        "--device", "cpu",			
     ]
     try:
         subprocess.run(command, check=True)