TunBERT / README.md
not-lain's picture
Update README.md
eb788d9
|
raw
history blame
804 Bytes
metadata
license: mit
datasets:
  - tsac
language:
  - ar

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') # make sure you are using the `return_tensors='pt'` parameter
output = model(**inputs)