model_cat_gen.py actualizedto run with jaime's data