Shardul-Ghuge
improved model after 15 epochs
812b2d1