The Colab notebook can be found here.