Update README.md

README.md

@@ -58,6 +58,28 @@ What is the Simplified Chinese translation of the sentence: The GRACE mission is
 <s>Assistant
 ```
 
+## Usage
+```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+
+tokenizer = AutoTokenizer.from_pretrained("nvidia/Riva-Translate-4B-Instruct")
+model = AutoModelForCausalLM.from_pretrained("nvidia/Riva-Translate-4B-Instruct").cuda()
+
+
+# Use the prompt template
+messages = [
+    {
+        "role": "system",
+        "content": "You are an expert at translating text from English to Simplified Chinese.",
+    },
+    {"role": "user", "content": "What is the Simplified Chinese translation of the sentence: The GRACE mission is a collaboration between the NASA and German Aerospace Center.?"},
+ ]
+tokenized_chat = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to(model.device)
+outputs = model.generate(tokenized_chat,  max_new_tokens=128, pad_token_id=tokenizer.eos_token_id)
+print(tokenizer.decode(outputs[0]))
+```
+
 ## Ethical Considerations:
 NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications.  When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.