Thanks for this ... but ....

[haalcala]

The model fails to come up with multiple suggestions if the input is "to" (it will always return "cover"). This is because before the tokenization, the string was processed to make sure there's only one word exists. As a side-effect, it captured only "to cover" which is incorrect. However "to" has a number of possible next word.