feat(model, train): improved architecture, overfit prevention, re-trained the model

feat(data): implemented data augmentation

initialized repo