Improved Colab notebook

[kwikwag]

🐛 Bug Report

The current Colab notebook references the old repo and it very cumbersome for no reason, including unnecessarily requiring mounting of Google Drive.

To Reproduce

Open the Colab link.

Expected behavior

It should use the new repo and be more user-friendly. It should not require mounting Google Drive. Ideally GPU should be optional.

Actual Behavior

See above.

Your Environment

Google Colab

My suggestion

I created a notebook with a user-friendly form, that doesn't require mounting Google Drive and that runs well on the CPU too. I would be flattered if you copied from me.