Generating Images from Text: with OpenAI’s DALL-E Python SDK

In this blog post, we will focus on DALL-E API, which is a neural network that generates images from textual descriptions.

OpenAI is a research laboratory consisting of the forerunners in Artificial Intelligence. The company has made significant progress in Natural Language Processing (NLP) and Computer Vision, among other fields. OpenAI has an API that developers can utilize to leverage their tools, like GPT-3, DALL-E, and Codex, and create custom solutions.

Steps to create an image using OpenAI API

• Get an API key: Before we can use OpenAI API, we need to get an API key. The process is straightforward, and we can do it by signing up on the OpenAI API website. We will get an API key once we sign up.

• Install the OpenAI Python module: We will use the OpenAI Python module to interact with the OpenAI API. To install the module, we can use pip by running the following command in our terminal or command prompt: pip install openai.

• Import the OpenAI module: After installing the module, we need to import it into our Python script or notebook. We can do it by running import openai.

• Set up authentication: To use the API, we need to set up our authentication by running the following command: openai.api_key = "YOUR_API_KEY". We need to replace the YOUR_API_KEY with the API key we got in step 1.

Send a request to DALL-E:

Code

The code below has all the above steps (using .env file)

import os
import openai

from dotenv import load_dotenv
load_dotenv()

openai.api_key = os.getenv("OPENAI_API_KEY")

response = openai.Image.create(
  prompt="a white glasses",
  n=1,
  size="1024x1024"
)
image_url = response['data'][0]['url']
print(image_url)

The code will generate a URL to a storage blob with the generated image.

If you would like to download the image to your local drive, add the code below.

import requests

response = requests.get(response['data'][0]['url'])
with open("apple.png", "wb") as f:
    f.write(response.content)

For more articles, visit the OpenAI category page.