bart-base-detox / README.md
dardem's picture
Create README.md
423d9ea
|
raw
history blame
317 Bytes

How to use

from transformers import BartForConditionalGeneration, AutoTokenizer
base_model_name = 'facebook/bart-base'
model_name = 'SkolkovoInstitute/bart-base-detox-10000-7'
tokenizer = AutoTokenizer.from_pretrained(base_model_name)
model = BartForConditionalGeneration.from_pretrained(model_name)