usage example did not work because of typo and torch not imported

Updated README

added checkpoints and readme file

Update README.md

initial commit