Updating Colab Notebook with example from 50 epoch model

50 epoch training w eval

Added a simplified version of the colab notebook, which doesnt have a clean env setup. This simplifies the setup process greatly at the cost of hurting reproducibility. Further research into ways to get a clean env on colab is required

Fixing colab after feedback

Update readme to include google colab setup + remove problematic packages from requirements.txt