kobart-text-style-transfer

kobart 모델에 Korean Smilestyle Dataset을 파인튜닝한 한국어 텍스트 스타일 변환 모델

예제

styles = ['문어체','구어체','안드로이드','아재','채팅',
    '초등학생','이모티콘','enfp','신사','할아버지','할머니','중학생',
    '왕','나루토','선비','소심한','번역기']

model = pipeline(
    'text2text-generation',
    model='heegyu/kobart-text-style-transfer'
)

def transfer_text_style(text, target_style, **kwargs):
  input = f"{target_style} 말투로 변환:{text}"
  out = model(input, max_length=64, **kwargs)
  print(text, target_style, out[0]['generated_text'], sep="->")

text = "반가운. 나는 6마리의 고양이를 소지하고 있다."
for style in styles:
  transfer_text_style(text, style)

결과

반가운. 나는 6마리의 고양이를 소지하고 있다.->문어체->안녕하세요. 저는 6마리의 고양이를 가지고 있습니다.
반가운. 나는 6마리의 고양이를 소지하고 있다.->구어체->안녕. 나는 6마리의 고양이를 가지고 있어.
반가운. 나는 6마리의 고양이를 소지하고 있다.->안드로이드->반갑다. 안드로이드. 6마리. 고양이. 보유.
반가운. 나는 6마리의 고양이를 소지하고 있다.->아재->안녕~~~~ 6마리의 고양이를 가지고 있네
반가운. 나는 6마리의 고양이를 소지하고 있다.->채팅->하이~ 6마리의 고양이 있음
반가운. 나는 6마리의 고양이를 소지하고 있다.->초등학생->ᄒᄋ 난 6마리 고양이 ᄏᄏ
반가운. 나는 6마리의 고양이를 소지하고 있다.->이모티콘->안녕!~()~ 난 6마리의 고양이를 가지고 있어 (皿)
반가운. 나는 6마리의 고양이를 소지하고 있다.->enfp->안녕!!~ 난 6마리의 고양이를 둬!
반가운. 나는 6마리의 고양이를 소지하고 있다.->신사->안녕하십니까, 저는 6마리의 고양이를 가지고 있습니다.
반가운. 나는 6마리의 고양이를 소지하고 있다.->할아버지->안녕하신가...나는 6마리의 고양이를 가지고 있구먼...
반가운. 나는 6마리의 고양이를 소지하고 있다.->할머니->염병 염병할 고양이 놈이여
반가운. 나는 6마리의 고양이를 소지하고 있다.->중학생->ᄒᄋ 난 6마리 고양이 키우는데
반가운. 나는 6마리의 고양이를 소지하고 있다.->왕->반갑소. 짐은 6마리의 고양이를 소유하고 있소.
반가운. 나는 6마리의 고양이를 소지하고 있다.->나루토->안녕하냐니깐! 난 6마리의 고양이를 가지고 있다니깐!
반가운. 나는 6마리의 고양이를 소지하고 있다.->선비->안녕하시오! 소생은 6마리의 고양이를 가지고 있소!
반가운. 나는 6마리의 고양이를 소지하고 있다.->소심한->안녕.... 난 6마리 고양이 있어..
반가운. 나는 6마리의 고양이를 소지하고 있다.->번역기->반가운, 나는 6마리의 고양이를 가지고 있다.