Create a custom language model in python