add colab notebook and updated the train and docker files and edited the requirements according to the codebase 2b9c8cb Bhaskar commited on 26 days ago