Create README.md

README.md

@@ -0,0 +1,32 @@
+---
+language:
+- ko
+---
+
+```
+from transformers import pipeline
+
+generator = pipeline(
+    'text-generation',
+    model="heegyu/ajoublue-gpt2-medium-dialog"
+)
+
+generation_args = dict(
+    max_new_tokens=32,
+    do_sample=True,
+    top_p=0.8
+)
+
+print(generator(
+    ["0 : **는 게임 좋아하니</s>1 :",
+    "0 : 어제 강남에서 살인사건 났대 ㅜㅜ 너무 무서워</s>1 : 헐 왜? 무슨 일 있었어?</s>0 : 사진보니까 막 피흘리는 사람있고 경찰들이 떠서 제압하고 난리도 아니었다던데??</s>1 :",
+    "0 : 자기야 어제는 나한테 왜 그랬어?</s>1 : 뭔 일 있었어?</s>0 : 어떻게 나한테 말도 없이 그럴 수 있어? 나 진짜 실망했어</s>1 : "],
+    **generation_args
+))
+
+[[{'generated_text': '0 : **는 게임 좋아하니</s>1 : 게임 좋아하죠'}], [{'generated_text': '0 : 어제 강남에서
+ 살인사건 났대 ㅜㅜ 너무 무서워</s>1 : 헐 왜? 무슨 일 있었어?</s>0 : 사진보니까 막 피흘리는 사람있고 경찰들 
+이 떠서 제압하고 난리도 아니었다던데??</s>1 : 헐 진짜 무섭다 ㅜㅜ 왜 그랬대?'}], [{'generated_text': '0 : 자
+기야 어제는 나한테 왜 그랬어?</s>1 : 뭔 일 있었어?</s>0 : 어떻게 나한테 말도 없이 그럴 수 있어? 나 진짜 실망
+했어</s>1 :  알았어. 그냥 서운한거 뿐이야.'}]]
+```