Update gen_represent_models.py