TunBERT / README.md
not-lain's picture
Update README.md
9cec50f
|
raw
history blame
711 Bytes
metadata
license: mit

Disclamer

I do not own, distribute, or take credits for this model, all copyrights belong to Instadeep under the MIT licence

how to load the model

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("not-lain/TunBERT")
model = AutoModelForSequenceClassification.from_pretrained("not-lain/TunBERT",trust_remote_code=True)

IMPORTANT : make sure to enable trust_remote_code=True

how to use the model

text = "[insert text here]"
inputs = tokenizer(text,return_tensors='pt')
output = model(**inputs)