Update split_multi-model_files.py