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)