Python으로 OpenAI API를 이용한 이미지 생성하기

OpenAI에서는 다양한 API를 제공하지만, 그 중에서도 이미지 생성을 위한 DALL-E API는 특히 주목받고 있습니다. DALL-E는 주어진 설명에 따라 이미지를 생성하는 딥러닝 모델로, 광범위한 주제로 이미지를 만들어낼 수 있습니다.

Generate Images

준비하기

Python을 이용해 OpenAI API에 접근하기 위해, 주의해야 할 사항이 있습니다.

코드 예제

다음은 주어진 텍스트 설명을 기반으로 이미지를 생성하는 Python 코드입니다.

from openai import OpenAI
import base64
client = OpenAI()

prompt = """
A children's book drawing of a veterinarian using a stethoscope to 
listen to the heartbeat of a baby otter.
"""

result = client.images.generate(
    model="gpt-image-1",
    prompt=prompt
)

image_base64 = result.data[0].b64_json
image_bytes = base64.b64decode(image_base64)

# Save the image to a file
with open("otter.png", "wb") as f:
    f.write(image_bytes)

위의 코드에서 `YOUR_API_KEY`를 본인의 API 키로 교체해 주세요. `prompt` 변수에 원하는 이미지의 설명을 입력하면, OpenAI API에 요청을 보내고 생성된 이미지의 URL을 출력합니다.

결과물 확인하기

코드를 실행하면, 콘솔에 생성된 이미지 URL이 출력됩니다. 이 URL을 브라우저에 붙여넣으면 생성된 이미지를 확인할 수 있습니다.

마무리

OpenAI의 DALL-E API를 이용하여 쉽고 빠르게 이미지를 생성할 수 있습니다. 다양한 설명을 사용하여 창의적인 이미지를 생성해 보시기 바랍니다. 더 많은 정보는 OpenAI의 공식 문서 를 참고하세요.