[Openai api] colab 환경에서 DALL.E 이미지 생성

openai 공식 튜토리얼

https://platform.openai.com/docs/quickstart/developer-quickstart

openai 공식 이미지 생성 튜토리얼

https://platform.openai.com/docs/guides/images/image-generation

openai 최신버전 다운로드

!pip install --upgrade openai

필요 라이브러리 불러오기, api key 넣기

from openai import OpenAI
import os
import sys
from IPython.display import display, Image

client = OpenAI(api_key='your api key')

이미지 생성 함수 작성

def image_generator(msg):
    response = client.images.generate(
                model="dall-e-3",
                prompt=msg,
                size="1024x1024",
                quality="standard",
                n=1,
                )
    image_url = response.data[0].url
    return image_url

프롬프트 작성 및 이미지 출력

message = "My friend's girlfriend cheated on him. I feel like he is very disappointed and confused."

image_url = image_generator(message)

display(Image(url=image_url))