Update README.md

README.md

@@ -8,6 +8,23 @@ pipeline_tag: translation
 
 This repository includes an MSA-to-LEV machine translation model. This model was used to curate dialectal benchmarks. The human post-edited benchmarks can be found<a href="https://huggingface.co/datasets/QCRI/AraDiCE" target="_blank" style="margin-right: 15px; margin-left: 10px">here.</a>
 
+## Sample Usage
+
+```python
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+
+tokenizer = AutoTokenizer.from_pretrained("QCRI/AraDiCE-msa-to-lev")
+model = AutoModelForSeq2SeqLM.from_pretrained("QCRI/AraDiCE-msa-to-lev")
+
+article = "يظهر سلف الأدب المكسيكي في آداب المستعمرات الأصلية في أمريكا الوسطى"
+inputs = tokenizer(article, return_tensors="pt")
+
+translated_tokens = model.generate(
+    **inputs, forced_bos_token_id=tokenizer.convert_tokens_to_ids("ajp_Arab"), max_length=30
+)
+translation = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
+print(translation)
+```
 ## License
 
 The model is distributed under the **Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License (CC BY-NC-SA 4.0)**. The full license text can be found in the accompanying `licenses_by-nc-sa_4.0_legalcode.txt` file.