Updated initialization

[akn-md]

This script should work with the updated DataLoader. The original was missing some imports, had a few outdated function calls and was using the same model for EF prediction and segmentation.

[tristan-deep]

Finding this lost PR solved quite some issues in making the code run for me. Maybe the authors can merge the PR.