I\'m trying to replicate the result of this github repo using Google Colab since I don\'t want to install all the requirements on my local machine and to take advantage of t
