improve train page to use proper models
improve live train model