Update the document for this project?

[eriche2016]

Hi, nice work, Can you tell me when will you update the document to run this code?

[neeharperi]

Thanks for your interest in our work! I'm currently in the process of refactoring the code, adding documentation, and pretrained models. I'll be done in the next few days. I'll add a note to this issue once this is complete.

[eriche2016]

Great, I cannot wait to try it.

[neeharperi]

Thanks for your patience! I've updated the documentation so that you can try out our codebase.