Added parameters for storing model in transformer for GUI runs.
Added NPE check so that tests still pass.