the example code provided has some small errors - this is a revision that I tested in colab, with the output commented at the bottom for users to verify
Thank you!
· Sign up or log in to comment
