Refactor load/create model from model_generator module to use single function for all objects