update readme

README.md

@@ -15,19 +15,15 @@ BART trained on Swedish data.
 ## Usage
 
 ```python
-from transformers import BartForConditionalGeneration, PreTrainedTokenizerFast
-from transformers import pipeline
+from transformers import BartForConditionalGeneration, PreTrainedTokenizerFast, AutoTokenizer
 
 model = BartForConditionalGeneration.from_pretrained("KBLab/bart-base-swedish-cased")
-tok = PreTrainedTokenizerFast(
-    tokenizer_file="tokenizer.json",
-    bos_token="<s>",
-    eos_token="</s>",
-    unk_token="<unk>",
-    pad_token="<pad>",
-    mask_token="<mask>",
-    cls_token="</s>",
-    sep_token="</s>",
+tok = AutoTokenizer.from_pretrained("KBLab/bart-base-swedish-cased")
+
+model.eval()
+
+input_ids = tok.encode(
+    "Jag har ätit en utsökt <mask> på restaurang vid <mask> .", return_tensors="pt"
 )
 
 model.eval()