Updating model

Updated BERT classifier on augmented of smaller classes

initial commit