Skip to main content

Image generation

The Image generation controller lets you create images based on text descriptions. This empowers developers to dynamically produce visuals for various applications.

Supported Providers

The framework supports different image generation models, allowing you to choose based on your needs: openai, stability.

Parameters

Specify these parameters to generate images:

  • api_key: Your API key or authentication token for the chosen provider.
  • provider: The name of the AI service provider (openai or stability).
  • image_input: This can be an instance of ImageModelInput containing the details of the image you want to generate, such as the prompt, dimensions, and model-specific parameters.

Example

Create an instance of the RemoteImageModel with your chosen provider and API key.

from intelli.controller.remote_image_model import RemoteImageModel

image_model = RemoteImageModel(api_key="your_api_key_here", provider="openai")

Prepare your image input, specifying the prompt and any other necessary parameters.

from intelli.model.input.image_input import ImageModelInput

# define the prompt and image parameters
prompt = "A futuristic city skyline at sunset, with flying cars."
image_input = ImageModelInput(
prompt=prompt,
width=1024,
height=1024,
model="dall-e-3"
)

# Generate the image
images = image_model.generate_images(image_input)