Modify script to automatically include the Github and Colab badges for notebooks

[nipunbatra]

See: https://github.com/nipunbatra/ml-teaching/blob/master/scripts/insert_badges.py

Eventually, we should create a pre-commit to ensure this is automatically done on all notebooks being pushed.