Add required imports to Generating Tokens example, and update with get_user_model()

The first commit fixes the imports, the second commit updates the User reference to a get_user_model() call. Take as much as you want. :)