Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
53 changes: 53 additions & 0 deletions Chat_gpt_api.ipynb
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"provenance": [],
"authorship_tag": "ABX9TyMzfOltuiIfV+7/NOvFLSPG",
"include_colab_link": true
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
},
"language_info": {
"name": "python"
}
},
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
},
"source": [
"<a href=\"https://colab.research.google.com/github/kongminhyuk/01/blob/patch-1/Chat_gpt_api.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"id": "_-7e9e-qjOFi"
},
"outputs": [],
"source": [
"from openai import OpenAI\n",
"\n",
"def get_completion(prompt, model=\"gpt-3.5-turbo\"):\n",
" client = OpenAI(api_key='API_KEY') #gpt api key 받아와서 입력\n",
" completion = client.chat.completions.create(\n",
" model=model,\n",
" messages=[{\"role\": \"user\", \"content\":prompt},])\n",
" response = completion.choices[0].message.content\n",
" return response\n",
"\n",
"prompt = input(\"검색하실 판례를 입력해주세요:\")\n",
"response = get_completion(prompt)\n",
"print(response)"
]
}
]
}
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,4 +9,10 @@
|ML Engineering|방가윤, 허정원|

## 사용 모델
[데이터](https://blog.lbox.kr/lbox-open)
[데이터](https://blog.lbox.kr/lbox-open)
MT5ForConditionalGeneration

## 프로젝트 개요
판례검색 Datasets으로 ibox를 이용하여 MT5ForConditionalGeneration 모델을 이용
학습 및 테스트 진행
모델과 chatgpt api 와 연결하여 사용
Loading