Update README.md

README.md

@@ -8,4 +8,14 @@ language:
 pipeline_tag: text2text-generation
 ---
 
-This is a huggingface port of the [PyTorch implementation of the original transformer](https://github.com/ubaada/scratch-transformer) model from 2017 introduced in the paper "[Attention Is All You Need](https://proceedings.neurips.cc/paper_files/paper/2017/file/3f5ee243547dee91fbd053c1c4a845aa-Paper.pdf)". This is the 65M parameter base model version trained to do English-to-German translations.
+This is a huggingface port of the [PyTorch implementation of the original transformer](https://github.com/ubaada/scratch-transformer) model from 2017 introduced in the paper "[Attention Is All You Need](https://proceedings.neurips.cc/paper_files/paper/2017/file/3f5ee243547dee91fbd053c1c4a845aa-Paper.pdf)". This is the 65M parameter base model version trained to do English-to-German translations.
+
+Usage:
+```python
+model = AutoModel.from_pretrained("ubaada/original-transformer", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("ubaada/original-transformer")
+text = 'This is my cat'
+output = model.generate(**tokenizer(text, return_tensors="pt", add_special_tokens=True, truncation=True, max_length=100))
+tokenizer.decode(output[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)
+# Output: ' Das ist meine Katze.'
+```