Sure! I think this is the correct notebook -

https://colab.research.google.com/drive/1IhS-RwgWXg3bII0hwseaugf4aAl83ouO?usp=sharing

Please note that in this notebook I directly loaded the uploaded dataset, so there is no code for the creation of data itself.