how we can run this code on google colab? I am having a connection error while trying to run this code on google colab.