youngjae99 / casanova

conversation helper using Korean GPT-2 model

Casanova

https://casanova-gpt2.netlify.app

이 말을 할까 말까 고민하는 당신을 위해 AI가 판단하고 도와줄께요!

이런 멘트가 담긴 톡을 호감있는 상대에게 보내도 괜찮은지 친구한테 물어보신 적 있다면 이제는 AI 연애 고수 Casanova에게 물어보세요.

당신의 멘트에 대한 점수와 함께, Casanova가 생각하는 다른 멘트를 추천해드립니다.

How it works

Casanova는 Ainize에서 제공하는 두 모델 API를 사용합니다. BERT기반 한국어 문장 공격성 분류 모델을 제공하는 SoongsilBERT-base-beep-deploy API를 통해 문장의 Hate or Offensive가 있는지를 바탕으로 멘트 점수(매력도)를 측정하고, KoGPT2와 Datasets from Instructions (DINO 🦕)를 이용하여 데이터를 생성하는 KoGPT2-DINO API를 이용해 유사하거나 동일한 의미를 갖는 문장을 AI가 더 나은 멘트를 추천해줍니다.

더 좋은 멘트 생성 결과를 보여주기 위한 fine-tuning된 모델은 차후 제공될 예정입니다 ( ´ ▽ ` )ﾉ

How to run

• Preset : git, docker, docker-compose should be installed on your server(or local server)
• git clone https://github.com/youngjae99/Casanova.git
• docker pull youngjae99/casanova
• docker-compose up -d --build

18.119.157.92 AWS EC2 link

Example input