did a 10M model and it doesn't work super great but its alright, add a full predict py for testing