Updated app.py

app.py

@@ -71,6 +71,8 @@ def run_nnunet_predict(nifti_file):
         "-tr", "nnUNetTrainer_8000epochs",
         "-device", "cuda"  # Explicitly use GPU
     ]
+    print("Files in /tmp/output:")
+    print(os.listdir(OUTPUT_DIR))
     try:
         subprocess.run(command, check=True)
 
@@ -86,14 +88,14 @@ def run_nnunet_predict(nifti_file):
             extract_middle_slice(input_path, input_slice_path)
             extract_middle_slice(new_output_file, output_slice_path)
 
+            # Return paths for the Gradio interface
             return input_slice_path, output_slice_path, new_output_file
         else:
             return "Error: Output file not found."
     except subprocess.CalledProcessError as e:
         return f"Error: {e}"
 
-# Gradio Interface
-interface = gr.Interface(
+# Gradio Interfaceinterface = gr.Interface(
     fn=run_nnunet_predict,
     inputs=gr.File(label="Upload FLAIR Image (.nii.gz)"),
     outputs=[