Update README.md

README.md

@@ -69,6 +69,25 @@ Given these potential pitfalls, and others not explicitly mentioned, it's essent
 ### License
 The model is licensed under [Research License](https://huggingface.co/microsoft/phi-1/resolve/main/Research%20License.docx).
 
+
+### Sample Code
+```python
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+torch.set_default_device('cuda')
+model = AutoModelForCausalLM.from_pretrained("microsoft/phi-1", trust_remote_code=True, torch_dtype="auto")
+tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1", trust_remote_code=True, torch_dtype="auto")
+inputs = tokenizer('''def print_prime(n):
+   """
+   Print all primes between 1 and n
+   """''', return_tensors="pt", return_attention_mask=False)
+
+outputs = model.generate(**inputs, max_length=500, eos_token_id=50256, pad_token_id=50256)
+text = tokenizer.batch_decode(outputs)[0]
+print(text)
+```
+
 ### Citation
 ```bib
 @article{gunasekar2023textbooks,