kinda works, predicts categories based on trained models