TarunSinghal's picture
Create README.md
26d7d1f
|
raw
history blame
334 Bytes
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)