shrunk size of training set and updated demo.py as a result