classifier_2 = BERTForClassification(bert_model=model, num_classes=6) | |
classifier_2.load_weights('/content/drive/MyDrive/FineTunning2/model2.1/model') | |
example_input = "Your work is very very bad" | |
encoded_example = tokenizer.encode(example_input, padding=True, truncation=True, return_tensors="tf") | |
p = classifier_2(encoded_example) |