Skip to content

Latest commit

 

History

History
111 lines (81 loc) · 4.9 KB

readme.md

File metadata and controls

111 lines (81 loc) · 4.9 KB

실시간 대화 요약 & 컨텐츠 추천 봇: TravelTalk

1. Introduction

image

Members

김현수 이성구 이현준 조문기 조익노
Git Badge Git Badge Git Badge Git Badge Git Badge
------------------------- ------------------------- ------------------------- ------------------------- -------------------------

| GitHub | GitHub | GitHub | GitHub | GitHub |

Contribution

  • 김현수   Front/Back-end 구현 • ElasticSearch 구현
  • 이성구   Product Manager(PM) • Dialogue summarization 고도화
  • 이현준   Dense retriever baseline 학습 및 평가 • ICT 응용기법 학습 및 평가 데이터 제작
  • 조문기   Front/Back-end 구현 • Dialogue summarization 및 metric 구현
  • 조익노   Data crawling scheduler 구현 • MongoDB 적용 • Retriever model 구현

Project Tree

|-- app
|   |-- assets
|   |-- src
|   |   |-- elastic
|   |   └-- models
|   |-- templates
|   |-- app.py
|   |-- config.py
|   └-- mongodb.py
|-- train
|   |-- summary
|   └-- retriever
└-- monstache
    └-- mongo-elastic.toml

2. Model

KoBART

image

Result

image

ColBERT

image

Result

image

3. Flow Chart

System Architecture

image

Pipeline

image

4. How to Use

Install Requirements

pip install -r requirements.txt

Getting Started

  • Summary model train
python ./train/summary/train.py
  • Retriever model train
python ./train/retriever/train.py
  • Run chat app with model server
python ./app.py
python ./src/models/summary_model.py
python ./src/models/retriever_model.py

5. Demo

image

6. Reference