Replace Model with PersistedModel.
Signed-off-by: Aris Kemper <aris.github@gmail.com>