corrected imports

updated README

made some changes

setup repo as python package

added README.md

updated requirements.txt

working demo, organized files

made some corrections

working training code with both the stages

first commit