Swap served kazan model for the trained diagram model

Initial commit