Update README.md

README.md

@@ -32,6 +32,27 @@ beomi/Yi-Ko-6B
 
 ## Uses
 
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model_id = "TeamUNIVA/Komodo_6B_v2.0.0"
+
+model = AutoModelForCausalLM.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+
+
+text = '''<|system|>
+당신은 사용자의 질문에 친절하게 답변을 하는 챗봇입니다.
+<|user|>
+안녕하세요?
+<|bot|>
+'''
+
+inputs = tokenizer(text, return_tensors="pt")
+outputs = model.generate(**inputs, max_new_tokens=20)
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
+
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
 ### Direct Use