snowseol/persona-translator

import torch
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM


repo_id = "snowseol/persona-translator"

tokenizer = AutoTokenizer.from_pretrained(repo_id)
model = AutoModelForSeq2SeqLM.from_pretrained(repo_id)

persona_list = {
    'informal': '구어체',
    'android': '안드로이드',
    'azae': '아재',
    'chat': '채팅',
    'choding': '초등학생',
    'emoticon': '이모티콘',
    'enfp': 'enfp',
    'gentle': '신사',
    'halbae': '할아버지',
    'halmae': '할머니',
    'joongding': '중학생',
    'king': '왕',
    'naruto': '나루토',
    'seonbi': '선비',
    'sosim': '소심한',
    'translator': '번역기'
}

tokenizer = AutoTokenizer.from_pretrained(repo_id)
model = AutoModelForSeq2SeqLM.from_pretrained(repo_id)


device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)
model.eval()

def generate_response(text: str, persona: str, max_length: int = 50):
    input_text = f"<{persona}>{text}"
    inputs = tokenizer(input_text, return_tensors="pt", padding=True, truncation=True).to(device)

    with torch.no_grad():
        output_ids = model.generate(
            input_ids=inputs["input_ids"],
            attention_mask=inputs["attention_mask"],
            max_length=max_length,
            num_beams=5,
            do_sample=False,
            early_stopping=True,
        )

    return tokenizer.decode(output_ids[0], skip_special_tokens=True)


if __name__ == "__main__":
    text = input("\n sentence to translate :").strip()
    for persona, kor_persona in persona_list.items():
        output = generate_response(text, kor_persona)
        print(f"{persona} :", output)

 sentence to translate : BBQ 샌드위치를 먹고 있습니다. 제가 제일 좋아하는 메뉴에요.
informal : 나는 샌드위치를 먹고 있어. 내가 제일 좋아하는 메뉴야.
android : 샌드위치. 섭취중. 안드로이드. 제일. 선호. 메뉴.
azae : 잉~BBQ 샌드위치 먹고 있어~ 내가 젤 좋아하는 메뉴야~
chat : BBQ 샌드위치 먹고 있음. 내가 제일 좋아하는 메뉴임
choding : BBQ 샌드위치 먹는 중 내가 젤 좋아하는 메뉴임
emoticon : 난 샌드위치 먹고 있어! (★★) 내가 제일 좋아하는 메뉴야! (★★)
enfp : 웅웅! 샌드위치 먹구 있어 ᄒᄒ 내가 젤 좋아하는 메뉴얌 ᄒᄒ
gentle : 샌드위치를 먹고 있습니다. 제가 제일 좋아하는 메뉴입니다.
halbae : 나는...BBQ 샌드위치를 먹고 있구먼...내가 제일 좋아하는 메뉴구먼...
halmae : 염병 샌드위치 먹고 있네 내가 젤 좋아하는 메뉴여
joongding : 샌드위치 먹는중 ᅳᅳ 내가 제일 좋아하는 메뉴임
king : BBQ 샌드위치를 먹고 있소. 짐이 제일 선호하는 메뉴요.
naruto : BBQ 샌드위치를 먹고 있다니깐! 내가 제일 좋아하는 메뉴라니깐!
seonbi : 샌드위치를 먹고 있소! 소생이 제일 좋아하는 메뉴인 것이오!
sosim : 샌드위치 먹고 있어.. 내가 제일 좋아하는 메뉴야..
translator : 나는 BBQ 샌드위치를 먹고 있다. 그것은 내가 가장 선호하는 메뉴입니다.

 sentence to translate : 안녕하세요. 세탁소에 맡긴 옷을 가지러 가는건데, 그리 급하지는 않습니다.
informal : 안녕. 세탁소에 맡긴 옷을 가지러 가는건데, 그리 급하지는 않아.
android : 반갑다. 세탁소. 맡긴 옷. 가지러 간다. 그러나. 그리.급하지. 않다.
azae : 안녕~ 세탁소에 맡긴 옷 가지러 가는건데, 그리 급하지는 않네
chat : 하이, 세탁소에 맡긴 옷 가지러 가는건데, 그리 급하지는 않음
choding : ᄒᄋ 세탁소에 맡긴 옷 가지러 가는건데, 그리 급하지는 않음
emoticon : 안녕! 세탁소에 맡긴 옷 가지러 가는건데, 그리 급하지는 않아 (⊙_⊙;)
enfp : 안녕~ 세탁소에 맡긴 옷 가지러 가는건뎅,, 그리 급하지는 않앙 ᄒᄒ
gentle : 안녕하십니까. 세탁소에 맡긴 옷을 가지러 가는건데, 그리 급하지는 않습니다.
halbae : 안녕하신가... 세탁소에 맡긴 옷을 가지러 가는 것이구먼...그렇게 급하지는 않구먼...
halmae : 염병하네 세탁소에 맡긴 옷을 가지러 가는건디 그리 급하지는 않어
joongding : ᄋ 세탁소에 맡긴 옷 가지러 가는건데 그리 급하지는 않음
king : 반갑소. 세탁소에 맡긴 옷을 가지러 가는 것이오. 그리 급하지는 않소.
naruto : 안녕하냐니깐! 세탁소에 맡긴 옷 가지러 가는 거라니깐! 그리 급하지는 않다니깐!
seonbi : 안녕하시오! 세탁소에 맡긴 옷을 가지러 가는 것이오! 그리 급하지는 않소!
sosim : 안녕.. 세탁소에 맡긴 옷 가지러 가는건데.. 그리 급하지는 않은 것 같아..
translator : 반가운. 세탁소에 맡긴 옷을 가지러 가는 것은 급하지 않다.

 sentence to translate : 의사 선생님은 뭐라고 하셨어요?
informal : 의사 선생님은 뭐라고 하셨어?
android : 의사. 선생님. 뭐라고. 하는가.
azae : 의사 선생님은 뭐라고 하셨어?
chat : 의사 선생님은 뭐라고 하셨음?
choding : 의사 선생님은 뭐라고 함?
emoticon : 의사 선생님은 뭐라고 하셨어? (⊙_⊙)?
enfp : 치과 선생님은 뭐라고 하셨엉??
gentle : 선생님은 뭐라고 하셨습니까?
halbae : 의사 선생님은 뭐라고 하신겐가?...
halmae : 의사 선생님은 뭐라고 하셨냐?
joongding : 의사 선생님은 뭐라고 했는데
king : 의사 선생님은 무엇을 말하였는가?
naruto : 의사 선생님은 뭐라고 하셨냐니깐!
seonbi : 선생님은 뭐라고 하셨소?
sosim : 의사 선생님은 뭐라고 하셨어 혹시..?
translator : 의사 선생님은 무엇이라고 하셨습니까?

 sentence to translate : 좋아보이는데요, 그러면 한번 게스트로 출연해볼게요.
informal : 좋아보이는데, 그러면 한번 게스트로 출연해볼게.
android : 좋음. 추측됨. 따라서. 게스트. 출연. 희망.
azae : 좋아보이네 그럼 한번 게스트로 출연해볼게
chat : 좋아보이는데, 그럼 한 번 게스트로 출연해볼게
choding : 좋아보이누 그럼 한번 게스트로 출연해봄
emoticon : 좋아보이네! 그럼 한 번 게스트로 출연해볼게! (★★)
enfp : 좋아보이넹!! 그럼 한번 게스트로 출연해볼게!!
gentle : 좋아보이네요. 그럼 한번 게스트로 출연해보겠습니다.
halbae : 좋아보이는구먼...그래서 한 번 게스트로 출연해보겠구먼...
halmae : 염병 좋아보이네 그럼 한번 게스트로 출연해볼게
joongding : 좋아보이네 그럼 게스트로 출연해볼게
king : 좋아보이오. 허면 한번 게스트로 출연해 보겠소.
naruto : 좋아보인다니깐! 그럼 한 번 게스트로 출연해보겠다니깐!
seonbi : 좋아보이오! 그럼 한번 게스트로 출연해보겠소!
sosim : 좋아보이는데.. 그럼 한번 게스트로 출연해볼게..
translator : 그것은 좋은. 따라서 나는 게스트로 출연한다.

 sentence to translate : 안 다치고 이렇게 오래 할 수 있었던게 축복인 것 같습니다. 저는 나무를 좋아하거든요.
informal : 다치고 이렇게 오래 할 수 있었던게 축복인 것 같아. 나는 나무를 좋아해.
android : 부상. 없음. 축복. 안드로이드. 나무. 선호.
azae : 안 다치고 이렇게 오래 할 수 있었던게 축복인 거 같네. 난 나무를 좋아하거든
chat : 안 다치고 이렇게 오래 할 수 있었던게 축복인 거 같음. 난 나무 좋아하거든
choding : 다치지 않고 이렇게 오래 할 수 있었던게 축복인 듯 난 나무 좋아함
emoticon : 다치지 않고 이렇게 오래 할 수 있었던게 축복인 것 같아 (*°▽°*) 난 나무 좋아하거든 (*°*)
enfp : 안 다치구 이렇게 오래 할 수 있었던게 축복인 거 가타!! 난 나무 좋아해 ᄒᄒ
gentle : 다치지 않고 이렇게 오래 할 수 있었던게 축복인 것 같습니다. 저는 나무를 좋아합니다.
halbae : 다치지 않고 이렇게 오래 할 수 있었던게 축복인 것 같구먼...난 나무를 좋아하는구먼...
halmae : 안 다치고 이렇게 오래 할 수 있었던게 축복인 거 같어. 난 나무 좋아하는디
joongding : 다치고 이렇게 오래 할 수 있었던게 축복임. 난 나무 좋아하는데
king : 다치지 않고 이렇게 오래 할 수 있었던 것은 축복인 것 같소. 짐은 나무를 선호하오.
naruto : 다치지 않고 이렇게 오래 할 수 있었던게 축복인 것 같다니깐! 난 나무를 좋아한다니깐!
seonbi : 다치지 않고 이렇게 오래 할 수 있었던게 축복인 것 같소! 소생은 나무를 좋아하오!
sosim : 안 다치고 이렇게 오래 할 수 있었던게 축복인 것 같아.. 난 나무 좋아해..
translator : 부상을 입지 않고 이렇게 오래 할 수 있었다는 것은 축복인. 나는 나무의 큰 애호가입니다.

 sentence to translate : 멋진 계획이네요. 이미 아시겠지만, 캐나다는 정말 아름다운 나라입니다.
informal : 멋진 계획이네. 이미 알겠지만, 캐나다는 정말 아름다운 나라야.
android : 계획. 훌륭함. 이미. 아는가. 캐나다. 아름다운 나라.
azae : 멋진 계획이네~ 이미 아시겠지만 캐나다는 정말 아름다운 나라여
chat : 멋진 계획이네. 이미 아시겠지만, 캐나다는 정말 아름다운 나라임
choding : ᆪᆪ 이미 아시겠지만 캐나다는 ᄅᄋ 아름다운 나라임
emoticon : 멋진 계획이네! (★★) 이미 아시겠지만, 캐나다는 정말 아름다운 나라야! (★★)
enfp : 멋진 계획이넹!! 이미 아시겠지만,, 캐나다는 진짜 아름다운 나라얌 ᄒᄒ
gentle : 멋진 계획이군요. 이미 아시겠지만, 캐나다는 정말 아름다운 나라입니다.
halbae : 멋진 계획이구먼...지금도 아시겠지만...캐나다는 정말 아름다운 나라이구먼...
halmae : 염병 지랄맞은 계획이네. 이미 알았지만 캐나다는 참말로 아름다운 나라여
joongding : ᄏ 이미 아시겠지만 캐나다는 아름다운 나라임
king : 멋진 계획이로오. 이미 알겠으나, 캐나다는 참으로 아름다운 나라요.
naruto : 멋지다니깐! 이미 아시겠지만 캐나다는 정말 아름다운 나라라니깐!
seonbi : 멋진 계획이오! 이미 아시겠지만, 캐나다는 정말 아름다운 나라인 것이오!
sosim : 멋진 계획인 것 같아.. 이미 아시겠지만 캐나다는 진짜 아름다운 나라야..
translator : 그것은 멋진 계획인. 이미 알겠지만, 캐나다는 정말 아름다운 나라입니다.

 sentence to translate : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되네요.
informal : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되네.
android : 따라서. 파이브 가이즈. 행버거. 섭취. 이유. 이해됨.
azae : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되네
chat : 그렇게 생각하면 파이브 가이즈 행버거 먹는 이유가 이해가 되네
choding : 그렇게 생각하면 파이브 가이즈 행버거 먹는 이유가 이해가 되네
emoticon : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되네 (★★)
enfp : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되넹 ᄒᄒ
gentle : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되네요.
halbae : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되는구먼...
halmae : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되네
joongding : 파이브 가이즈 행버거 먹는 이유가 이해가 안되네
king : 그리 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되는구려.
naruto : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 된다니깐!
seonbi : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 되는 것이오!
sosim : 그렇게 생각하면 파이브 가이즈 행버거 먹는 이유가 이해가 되는 것 같아..
translator : 그렇게 생각하면 파이브 가이즈 행버거를 먹는 이유가 이해가 된다.

 sentence to translate : 여기 수족관은 볼 거리가 정말 많군요.
informal : 수족관은 볼 거리가 정말 많군.
android : 수족관. 볼 거리. 많음.
azae : 어기 수족관은 볼 거리가 정말 많네
chat : 수족관은 볼 거리가 진짜 많네
choding : 수족관은 볼 거리가 ᄅᄋ 많네
emoticon : (*°▽°*)여기 수족관은 볼 거리가 정말 많네 (★★)
enfp : 웅! 수족관은 볼 거리가 진짜 많넹 ᄒᄒ
gentle : 네. 수족관은 볼 거리가 정말 많군요.
halbae : 그렇구먼...여기 수족관은...볼 거리가 정말 많구먼...
halmae : 니기럴 수족관은 볼 거리가 염병하게 많네
joongding : 수족관은 볼 거리가 너무 많네
king : 그렇소. 수족관은 볼 거리가 참으로 많구려.
naruto : 그렇다니깐! 수족관은 볼 거리가 정말 많다니깐!
seonbi : 수족관은 볼 거리가 정말 많소!
sosim : 수족관은 볼 거리가 진짜 많은 것 같아..
translator : 그런, 수족관은 볼 거리가 아주 많다.

 sentence to translate : 그리고 스키장 무료 입장권도 받았어요.
informal : 그리고 스키장 무료 입장권도 받았어.
android : 스키장. 무료. 입장권. 수령.
azae : 글고 스키장 무료 입장권도 받았어
chat : 글고 스키장 무료 입장권도 받음
choding : 글고 스키장 무료 입장권도 받음
emoticon : 그리고 스키장 무료 입장권도 받았어! (★★)
enfp : 글구 스키장 무료 입장권도 받았엉 ᄒᄒ
gentle : 그리고 스키장 무료 입장권도 받았습니다.
halbae : 그리고...스키장 무료 입장권도 받았구먼...
halmae : 염병 스키장 무료 입장권도 받았네
joongding : 스키장 무료 입장권도 받음
king : 그리하여 스키장 무료 입장권도 받았소.
naruto : 그리고 스키장 무료 입장권도 받았다니깐!
seonbi : 그리고 스키장 무료 입장권도 받았소!
sosim : 그리고 스키장 무료 입장권도 받았어..
translator : 나는 스키장 무료 입장권을 받았다.

 sentence to translate : 지금 물어봤는데, 동생도 시간이 안된다고 하네요.
informal : 지금 물어봤는데, 동생도 시간이 안된다고 하네.
android : 확인. 동생. 시간. 안된다.
azae : 나도 물어봤는데, 동생도 시간이 안된다고 하네
chat : 나도 물어봤는데 동생도 시간이 안된다고 하더라
choding : 나도 물어봤는데 동생도 시간이 안된대
emoticon : 나도 물어봤는데 동생도 시간이 안된다고 하더라...(⊙_⊙;)
enfp : 지금 물어봤는뎅,, 동생두 시간이 안된다구 하넹,,
gentle : 지금 물어봤습니다. 동생도 시간이 안된다고 하더군요.
halbae : 나도...지금 물어봤는디...동생도 시간이 안된다고 하더구먼...
halmae : 시방 물어봤는디, 동생도 시간이 안된대
joongding : 나도 물어봤는데 동생도 시간 안된다고 하더라
king : 금일 물어봤는데, 동생 또한 시간이 안된다고 하였소.
naruto : 지금 물어봤는데 동생도 시간이 안된다고 하더라니깐!
seonbi : 지금 물어봤소! 동생도 시간이 안된다고 하오!
sosim : 나도 물어봤는데 동생도 시간이 안된다고 하더라구..
translator : 나는 그것에 대해 물었고, 동생은 시간이 없다고 대답했다.