updated config defaults

chore: add trained weights using Git LFS

improve modularity of the model

refactiring rest of code base and adding checkpoints