Update pip install dependencies

[rodericj]

There was a version discrepancy when running this notebook in google collab. I've unpinned a few of the dependencies to allow pip to do its magic.