Gradio와 하이퍼클로버X 예제

필요한 모듈 설치하기

!pip install openai
!pip install gradio

키 정의하기

import os
os.environ["CLOVA_API_KEY"] = "당신의 CLOVA Studio API Key를 입력하세요"

Gradio에서 하이퍼클로바X 모델 호출하기

import gradio as gr
from openai import OpenAI

client = OpenAI(
    api_key = os.environ["CLOVA_API_KEY"],
    base_url="https://clovastudio.stream.ntruss.com/v1/openai"
)

def generate_response(prompt_text, history):
    completion = client.chat.completions.create(
        model="HCX-005",
        messages=[
            {"role": "user", "content": prompt_text}
        ]
    )

    response = completion.choices[0].message;
    return response.content

gr.ChatInterface(generate_response, title="나의 하이퍼클로바X 챗봇").launch()