Repo and codelab drift

[dstack4273]

Hello Google Codelab team,

This repo has drifted a bit to the point where it needs a lot of fiddling to get functional It looks like some of the imports and library versions are dated as well. It would be super helpful for future explorers to make things a bit simpler to get up and running.