Update README.md

README.md

@@ -1,3 +1,11 @@
+---
+language:
+- ko
+tags:
+- gpt2
+license: cc-by-nc-sa-4.0
+---
+
 ## Example
 ```python
 import torch
@@ -20,7 +28,7 @@ model = AutoModelForCausalLM.from_pretrained(
 model.eval()
 
 prompt = "석양이 보이는 경치"
-wrapped_prompt = f"@{prompt}@"
+wrapped_prompt = f"@{prompt}@<usr>\n"
 with torch.no_grad():
     tokens = tokenizer.encode(wrapped_prompt, return_tensors="pt").to(device="cuda")
     gen_tokens = model.generate(
@@ -36,4 +44,10 @@ with torch.no_grad():
     generated = tokenizer.decode(gen_tokens[0][len(tokens[0]):])
  
 print(generated)
+# 해가 지고 있을 무렵
+# 나는 석양을 보러 간다
+# 붉은 하늘과 하얀 구름이 나를 반겨줄 것 같아서리
+# 하지만 내가 본 해는 저물어만 가고
+# 구름마저 자취를 감춘 어둠만이 남아있을 뿐이네
+# 내가 탄 배는 보이지도 않고
 ```