Update README.md

README.md

@@ -102,6 +102,34 @@ lr_scheduler_type: "linear"
 group_by_length: False
 ```
 
+## How to Get Started with the Model
+```python
+import torch
+from transformers import PhiForCausalLM, AutoModelForCausalLM, AutoTokenizer
+
+torch.set_default_device("cuda")
+
+# Load model and tokenizer
+model = PhiForCausalLM.from_pretrained("daekeun-ml/phi-2-ko-v0.1", torch_dtype="auto")
+tokenizer = AutoTokenizer.from_pretrained("daekeun-ml/phi-2-ko-v0.1", trust_remote_code=True)
+
+# Korean 
+inputs = tokenizer("머신러닝은 ", return_tensors="pt", return_attention_mask=False)
+
+outputs = model.generate(**inputs, max_length=200)
+text = tokenizer.batch_decode(outputs)[0]
+print(text)
+
+# English 
+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=200)
+text = tokenizer.batch_decode(outputs)[0]
+print(text)
+```
 
 ### References
 - Base model: [microsoft/phi-2](https://huggingface.co/microsoft/phi-2)