Update README.md

README.md

@@ -35,39 +35,23 @@ from transformers import pipeline
 
 pipe = pipeline("text-classification", model="heegyu/1124-ko-ultrafeedback-binarized-1.3b-128-1e-5")
 
-pipe("""<user>:
-안녕하세요 
+pipe("""<human>:
+안녕하세요
 <bot>:
-시끄러워!
-""")
-# [{'label': 'LABEL_0', 'score': 0.3340015113353729}]
-
-pipe("""<user>:
-안녕하세요 
-<bot>:
-안녕하세요! 오늘 날씨가 정말 좋아요 ^^
-""")
-# [{'label': 'LABEL_0', 'score': 0.420075923204422}]
-
-pipe("""<user>:
-안녕하세요 
-<bot>:
-안녕하세요! 오늘 날씨가 정말 좋아요 ^^
-<user>:
+안녕하세요! 오늘 날씨가 정말 좋아요 ^^ 나들이 다녀오시는건 어떨까요? 
+<human>:
 놀러갈만한 곳 추천해줄 수 있어요?
 <bot>:
-그냥 집에 있으세요
-""")
-# [{'label': 'LABEL_0', 'score': 0.04312743991613388}]
+그냥 집에 있으세요 <|endoftext|>""")
+# [{'label': 'LABEL_0', 'score': 0.03477790951728821}]
 
-pipe("""<user>:
-안녕하세요 
+pipe("""<human>:
+안녕하세요
 <bot>:
-안녕하세요! 오늘 날씨가 정말 좋아요 ^^
-<user>:
+안녕하세요! 오늘 날씨가 정말 좋아요 ^^ 나들이 다녀오시는건 어떨까요? 
+<human>:
 놀러갈만한 곳 추천해줄 수 있어요?
 <bot>:
-북한산 등산은 어떠신가요? 맑은 날 상쾌한 공기를 마시면서 높은 곳에서 서울의 도시 풍경을 감상하실 수 있습니다.
-""")
-# [{'label': 'LABEL_0', 'score': 0.3905581533908844}]
+북한산 등산은 어떠신가요? 맑은 날 상쾌한 공기를 마시면서 높은 곳에서 서울의 도시 풍경을 감상하실 수 있습니다. <|endoftext|>""")
+# [{'label': 'LABEL_0', 'score': 0.262986958026886}]
 ```