how to save multiple models in python