Updated app.py

app.py

@@ -37,9 +37,6 @@ 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
@@ -51,9 +48,10 @@ def run_nnunet_predict(nifti_file):
         "-o", OUTPUT_DIR,
         "-d", "004",                  # Dataset ID
         "-c", "3d_fullres",           # Configuration
-        "-tr", "nnUNetTrainer_8000epochs",
-        "-device", "cpu",			
+        "-tr", "nnUNetTrainer_8000epochs",			
     ]
+    print("Files in /tmp/output:")
+    print(os.listdir(OUTPUT_DIR))
     try:
         subprocess.run(command, check=True)
         # Get the output file