Update README.md

README.md

@@ -2,7 +2,7 @@ kakao brain에서 공개한 kogpt 6b model('kakaobrain/kogpt')을 fp16으로 저
 
 ### 카카오브레인 모델을 fp16으로 로드하는 방법
 
-```
+```python
 import torch
 from transformers import GPTJForCausalLM
 
@@ -11,7 +11,7 @@ model = GPTJForCausalLM.from_pretrained('kakaobrain/kogpt', cache_dir='./my_dir'
 
 ### fp16 모델 로드 후 문장 생성
 
-```
+```python
 import torch
 from transformers import GPTJForCausalLM, AutoTokenizer