Skip to content

Prometheus-AI-Project/2024_2_Basic_Study_LLM-PCR

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

📊 Product Comparision & Recommendation

온라인에서 판매되는 다양한 제품 정보를 비교하고 분석해주는 AI 챗봇 서비스
사용자가 특정 상품에 대해 질문을 하면, AI 챗봇은 비슷한 제품을 비교분석하고 추천해주는 답변을 제공합니다.


⚙️ Pipeline


사용자로부터 특정 제품에 대해서 비슷한 제품을 비교하고 분석받고 싶다는 요청이 들어오면, RAG 기법을 활용하여 비슷한 제품의 설명을 담고 있는 웹 사이트를 찾은 후 자동으로 크롤링을 진행합니다.

이후 크롤링 된 결과를 전처리하고 LLM을 통해 압축된 중요한 정보가 커스텀 프롬프트와 함께 AI 기반 챗봇에게 전달되면 제품을 분석하여 사용자에게 답변을 제공합니다.


⚡️ Quick Start

python app.py

실행화면


User Question 입력창에 비교 및 분석을 원하시는 제품명과 함께 질문한 후, 하단의 Let me Know 버튼을 눌러주세요.

Example:

삼성 갤럭시 버즈3 pro와 비슷한 가격대의 제품을 비교 분석하여 추천해줘

✅ Envirionment Setting

1️. 가상환경 생성

conda create -n pcr python=3.11
conda activate pcr

2️. 필요 라이브러리 설치

pip install -r requirements.txt

📁 Dataset Construction

python dataset.py

Langchain과 GPT API를 활용하여 자체적으로 질문-정보-답변 데이터셋을 구축합니다.


🚀 Train & Inference

1️. Train

python main.py --train=1

LoRA 기법을 활용하여 Instruction Tuning을 진행하여 모델이 비교분석 및 추천하는 답변을 더욱 잘 생성하게 만듭니다.

2️. Inference

python main.py --train=0

🛠️ Tools & Technologies

Image


👥 Contributions

이름 김도현 김채린 김천지 유지혜
Github ID @Dohyeon-Kim1 @KimChaeRin-0601 @CheonjiKim @jihyeyu33

About

온라인에서 판매되는 다양한 제품 정보를 비교하고 분석해주는 AI 챗봇 서비스

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%