Update README.md with screenshots

Add screenshot images

Create README.md

Change the name of the environment in environment.yml

Add environment.yml

Factor out train and test functions

First draft of tensorboard demo

Add .gitignore