EOFError: Ran out of input in Google Colab for a Sentiment Analysis model

Question

I am trying a sentiment analysis with the IMDB dataset using Pytorch. Running it on google colab gives me an error

EOFError                                  T