[AI] 파이썬으로 GPT 사용하기

안녕하세요.

오늘은 파이썬으로 OpenAI의 GPT3를 간단하게 사용하는 방법에 대해 알아보겠습니다.

 

※ 과거 ChatGPT에 대해 포스팅한 글이 있으니 관심 있으신 분들은 참고하시기 바랍니다.

2022.12.14 - [Programming/AI] - [OpenAI] ChatGPT 인공지능 챗봇

 

 OpenAI API & 파이썬 활용

▣ OpenAI API

: 사이트 바로가기

- OpenAI에서 만든 GPT3를 사용하기 위해서는 API key(무료 제공)를 받아야 합니다.

- 다음 사이트에서 회원가입 후 API key를 받으면 됨.

- 아래와 같이 USER > API Keys에서 개인 API key를 받으면 됨. (복사 후 개별 저장 필수)

 

: Python OpenAI GPT3 사용하기

- 현재 OpenAI GPT3는 한국어 버전은 느려서 영어로 진행해보겠음.

- openai install 실행

pip install openai

- 파이썬에서 openai chatgpt에 프롬프트 입력해서 출력 받기

- 프롬프트에는 워렌 버핏의 포트폴리오에 대한 질문 넣기

import openai

openai.api_key = "openai API KEY 값"

prompt = "Tell me Warren Buffet's portfolio"

response = (openai.Completion()).create(
        engine="text-davinci-003",
        prompt=prompt,
        temperature=0,
        max_tokens=100,
        top_p=1,
        frequency_penalty=0.0,
        presence_penalty=0.0,
        best_of=1,)

print(response.choices[0].text.strip())

- GPT3 결과 보기

Warren Buffett's portfolio is composed of a variety of stocks, bonds, and other investments. His largest holdings include Apple, Bank of America, Coca-Cola, American Express, Wells Fargo, and Kraft Heinz. He also owns a variety of other stocks, including IBM, Goldman Sachs, and Walmart. He also owns a variety of bonds, including U.S. Treasury bonds and municipal bonds.

 

: 간단한 Q&A 챗봇 만들기

- 프롬프트에 "종료"라고 칠 때까지 질문을 하면 답하게 끔 코드 구성

while True:
    question = input("Q:")
    if question == '종료' or question == 'exit':
        break

    response = (openai.Completion()).create(
        engine="text-davinci-003",
        prompt=question,
        temperature=0,
        max_tokens=300,
        top_p=1,
        frequency_penalty=0.0,
        presence_penalty=0.0,
        best_of=1,
    )

    print(response.choices[0].text.strip())

ChatGPT 챗봇 결과

 

- openai 함수에 들어가는 모델과 변수에 대한 자료는 openai 사이트에 자세히 나와 있으니 관심 있으신 분들은 아래 사이트들을 참고하시기 바랍니다.

• OpenAI API Reference
• OpenAI GPT3 Models