Fix notebook dying due to inactivity on Google Colab (ch 4, nb 7)

[Ethan0456]

The Notebook dies due to inactivity on Google Colab, when running cells which takes 30 minutes or more. Suspected reasons may be due to unstable internet connection of user or an issue from Google Colab.

Screenshot (Progress stops and the notebook needs to be rerun)