Just to let others know if they are facing the same issue. You can try in Kaggle if you face this issue in colab.