Create README.md

README.md

@@ -0,0 +1,11 @@
+```
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large")
+model = AutoModelForSeq2SeqLM.from_pretrained("vadis/bart_scitldr", use_auth_token=True)
+
+text = "Abstract of a paper."
+batch = tok(text, return_tensors="pt")
+generated_ids = model.generate(batch["input_ids"])
+print(tok.batch_decode(generated_ids, skip_special_tokens=True))
+```