diff --git a/scripts/ML/BERT.py b/scripts/ML/BERT.py index ec4381c9c2fd5c07a9a0707cd08e3fe298049cfa..126d4e7e12f5e878b176abe8a1c6190122057aaf 100644 --- a/scripts/ML/BERT.py +++ b/scripts/ML/BERT.py @@ -5,10 +5,12 @@ class BERT: model_name = 'bert-base-multilingual-cased' def __init__(self, path): - print('Loading BERT tools…') - self.tokenizer = BertTokenizer.from_pretrained(BERT.model_name) - print('âœ”ï¸ tokenizer') self.device = get_device() + print('Loading BERT tools') + print(' - tokenizer', end='') + self.tokenizer = BertTokenizer.from_pretrained(BERT.model_name) + print('\râœ”ï¸ tokenizer') + print(' - classifier', end='') bert = BertForSequenceClassification.from_pretrained(path) self.model = bert.to(self.device.type) - print('âœ”ï¸ classifier') + print('\râœ”ï¸ classifier')