Skip to main content

Use DALL·E 2 in Python

Project description

Get Access

labs.openai.com/waitlist

Usage

pip install dalle2
from dalle2 import Dalle2

dalle = Dalle2("sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxx")
generations = dalle.generate("portal to another dimension, digital art")

print(generations)
✔️  Task created with ID: task-f77yxcsdf3OEm and PROMT: portal to another dimension, digital art
⌛ Waiting for task to finish .. 
🙌 Task completed!

[
  {
    'id': 'generation-sCnERSYDPP0Zu14fsdXEcKmL',
    'object': 'generation',
    'created': 1553332711,
    'generation_type': 'ImageGeneration',
    'generation': {
      'image_path': 'https://openailabsprodscus.blob.core.windows.net/private/user-hadpVzldsfs28CwvEZYMUT/generations/generation...'
    },
    'task_id': 'task-nERkiKsdjVCSZ50yD69qewID',
    'prompt_id': 'prompt-2CtaLQsgUbJHHDoJQy9Lul3T',
    'is_public': false
  },
  {
    'id': 'generation-hZWt2Nasrx8R0tJjbaROfKVy',
    'object': 'generation',
    'created': 1553332711,
    'generation_type': 'ImageGeneration',
    'generation': {
      'image_path': 'https://openailabsprodscus.blob.core.windows.net/private/user-hadpVzldsfs28CwvEZYMUT/generations/generation...'
    },
    'task_id': 'task-nERkiKhjasdSZ50yD69qewID',
    'prompt_id': 'prompt-2CtaLasdUbJHHfoJQy9Lul3T',
    'is_public': false
  },
  # 4 more ... 
]

or download all generations

from dalle2 import Dalle2

dalle = Dalle2("sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxx")
generations = dalle.generate_and_download("portal to another dimension, digital art")
✔️  Task created with ID: task-f77sayxcSGdfOEm and PROMT: portal to another dimension, digital art
⌛ Waiting for task to finish .. 
🙌 Task completed!
Download to directory: C:\Users\pc\dalle2
✔️  Downloaded:  generation-fAq4Lyxcm7pQVDBQEWJ.jpg
✔️  Downloaded:  generation-zqfBC3yyxcPXRlW6zLP.jpg
✔️  Downloaded:  generation-soR3ryxcoeixzdyHG.jpg
✔️  Downloaded:  generation-lT5L4yxc2DOiGRwJi.jpg
✔️  Downloaded:  generation-01DvPryxcq2BX1NOREL.jpg
✔️  Downloaded:  generation-AAs4yxcczH1vl19FidR.jpg

or generate a specific amount

from dalle2 import Dalle2

dalle = Dalle2("sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxx")
generations = dalle.generate_amount("portal to another dimension", 12) # Every generation has batch size 6 -> amount % 6 == 0 works best

print(generations)
✔️  Task created with ID: task-lm0V4nZasgAFasd7AsStE67 and PROMT: portal to another dimension OVERALL: 1/ 2
⌛ Waiting for task to finish .. 
➕  Appended new generations to all_generations
✔️  Task created with ID: task-WcetZOHt8asdvHb433gi and PROMT: portal to another dimension OVERALL: 2/ 2
⌛ Waiting for task to finish .. 
➕  Appended new generations to all_generations
🙌 Task completed!

-> [list]

Test In Colab

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

dalle2-1.0.6.tar.gz (4.0 kB view hashes)

Uploaded Source

Built Distribution

dalle2-1.0.6-py3-none-any.whl (4.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page