[파이썬] kogpt2 이용하기

https://developers.kakao.com/product/kogpt

Kakao Developers

 

 

https://developers.kakao.com/docs/latest/ko/kogpt/rest-api

Kakao Developers

 

import requests
import json

# kogpt REST API키값
api_key = ""

def generate_summary(input_text,max_tokens=64,temperature=0.7,top_p=0.8):

    
    # API 요청 주소
    url = "https://api.kakaobrain.com/v1/inference/kogpt/generation"

    # API 요청 데이터
    data = {
        "prompt": input_text,
        "max_tokens": 128,
        "top_p":0.7
        
    }

    # API 요청 헤더
    headers = {
            'Authorization': 'KakaoAK ' + api_key,
            'Content-Type': 'application/json'
    }

    # API 요청 보내기
    response = requests.post(url, headers=headers, data=json.dumps(data))

    # API 응답 결과 파싱
    result = json.loads(response.content.decode('utf-8'))


    return result

input_text = ""
summary = generate_summary(input_text)
print(summary)