Things in tf_models are getting full of code redundancies. A good refactor would make things more clear.