Update app.py

app.py

@@ -116,8 +116,8 @@ def run_nnunet_predict(nifti_file):
             # Extract and save 2D slices
             input_slice_path = os.path.join(OUTPUT_DIR, f"{base_filename}_input_slice.png")
             output_slice_path = os.path.join(OUTPUT_DIR, f"{base_filename}_output_slice.png")
-            extract_middle_slice(input_path, input_slice_path)
-            extract_middle_slice(new_output_file, output_slice_path)
+            extract_middle_slices(input_path, input_slice_path)
+            extract_middle_slices(new_output_file, output_slice_path)
 
             # Return paths for the Gradio interface
             return new_output_file, input_slice_path, output_slice_path