gogamza
/

kobart-summarization

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

gogamza commited on May 12, 2021

Commit

6b2ed56

•

1 Parent(s): 5b5a14a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ license: MIT
 from transformers import PreTrainedTokenizerFast
 from transformers import BartForConditionalGeneration
-tokenizer = PreTrainedTokenizerFast(
     'gogamza/kobart-summarization',
     bos_token='<s>', eos_token='</s>', unk_token='<unk>', pad_token='<pad>', mask_token='<mask>')

 from transformers import PreTrainedTokenizerFast
 from transformers import BartForConditionalGeneration
+tokenizer = PreTrainedTokenizerFast.from_pretrained(
     'gogamza/kobart-summarization',
     bos_token='<s>', eos_token='</s>', unk_token='<unk>', pad_token='<pad>', mask_token='<mask>')