Trying out BERT embeddings and tokenization as a replacement for one-hot encoding. Implement and compare results with the baseline model.