Skip to content

[제 12회 투빅스 컨퍼런스] Meet Persona Through AI - 영화 속 캐릭터 대화 챗봇

Notifications You must be signed in to change notification settings

datamarket-tobigs/MPTI

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

94 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

📽MPTI💬

youtube googledrive


MPTIMeet Persona Through AI의 줄임말로 영화 속 캐릭터와의 대화형 챗봇 서비스입니다.

MPTI_Sherlock은 DialoGPT를 사용하여, MPTI_Ironman은 GPT2를 사용하여 만들었습니다.

서비스에 대한 자세한 내용은 그림 위의 링크를 참고해주세요.

Data

학습에 사용된 데이터는 대본을 바탕으로 구성되었으며, 상업적으로 시용할 의도가 전혀 없음을 밝힙니다:)

각각의 챗봇에 사용한 데이터의 형식이 서로 다릅니다. 데이터의 형식은 개별 README를 통해 확인할 수 있습니다.

Usage

모델의 용량 문제로 인해, 16GB 이상의 GPU 환경이나 Colab에서 GPU를 사용하기를 적극 권장합니다. Colab 환경의 경우, [런타임] - [런타임 유형 변경] - [GPU]로 설정을 바꿔주세요.

1. Installation

git clone https://github.com/TobigsConf-1415/MPTI.git
cd MPTI
pip install -r requirements.txt

2. Train

각 모델을 아래의 순서대로 학습시켜주세요

cd DialoGPT
python main.py
cd TransferTransfo
python -m spacy download en
python train.py

3. Inference

python demo.py

Result


Demo

Sherlock Iron Man

Reference

Contributor 🕵️‍♂️


Seyoung Jung

Jaeyoon Jeong

Taehee Kim

Sumin Lee

Junhyeok Cho

JinKyoung Hwangbo

About

[제 12회 투빅스 컨퍼런스] Meet Persona Through AI - 영화 속 캐릭터 대화 챗봇

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 91.5%
  • Jupyter Notebook 7.2%
  • Shell 1.3%