Sweep: improve upon this code base so that anyone can install using pip and train and inference on their own data.

[hemangjoshi37a]

I want to make changes in the way that people can install this repo using pip , and in this people/users can use their own data.