dalle2 in python

Openai a une API officielle maintenant

Installation

pip install dalle2

Usage

Installation

1. Allez sur https://openai.com/dall-e-2/
2. Créer un compte Openai
3. Allez sur https://lab.openai.com/
4. Ouvrez l'onglet réseau dans les outils de développeur
5. Tapez une invite et appuyez sur "Générer"
6. Recherchez Fetch sur https://labs.openai.com/api/labs/tasks
7. Dans l'en-tête de demande, recherchez l'autorisation puis obtenez le jeton de porteurs

 from dalle2 import Dalle2
dalle = Dalle2 ( "sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxx" ) # your bearer key 

Générer des images

 generations = dalle . generate ( "portal to another dimension, digital art" )
print ( generations )

 ✔️ Task created with ID: task-xsuhOthvBXLEjddn3ynyiiOR
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? 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
  },
  # 3 more ... 
]

Générer des images et les télécharger

 file_paths = dalle . generate_and_download ( "portal to another dimension, digital art" )

 ✔️ Task created with ID: task-xsuhOthvBXLEjddn3ynyiiOR
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!
✔️ Downloaded: C:...generation-XySidj4N8EN6Ok9ed15BZ2bs.webp
✔️ Downloaded: C:...generation-IK3UdxDz77FA5SLKpQPIITdU.webp
✔️ Downloaded: C:...generation-uNejKBXz1z6EQxJAT9pAZbof.webp
✔️ Downloaded: C:...generation-Ol1wEqNprf34vNohmJz0iUiE.webp

Générer un nombre spécifique d'images

 generations = dalle . generate_amount ( "portal to another dimension" , 8 ) # Every generation has batch size 4 -> amount % 4 == 0 works best

# download images
file_paths = dalle . download ( generations )
print ( file_paths )

 ✔️ Task created with ID: task-lm0V4nZasgAFasd7AsStE67
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!
✔️ Task created with ID: task-WcetZOHt8asdvHb433gi
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!

Générer des images à partir d'un fichier masqué

Dall · E prend en charge une API "d'insistance" qui remplit des parties transparentes d'une image. Le site Web fournit un outil pour peindre une image existante pour indiquer quelles pièces vous souhaitez être transparente. Cet appel de package Python suppose que l'image que vous fournissez a déjà été traitée pour avoir des pièces transparentes.

 # make the right half of a saved image transparent
from PIL import Image , ImageDraw

image = Image . open ( 'my_image.png' )
m , n = image . size

area_to_keep = ( 0 , 0 , m // 2 , n )
image_alpha = Image . new ( "L" , image . size , 0 )
draw = ImageDraw . Draw ( image_alpha )
draw . rectangle ( area_to_keep , fill = 255 )

image_rgba = image . copy ()
image_rgba . putalpha ( image_alpha )
image_rgba = image_rgba . resize (( 1024 , 1024 )) # image must be square and 1024x1024
image_rgba . save ( 'image_with_transparent_right_half.png' )

# ask DALL·E to fill-in the transparent right half
generations = dalle . generate_from_masked_image (
    "portal to another dimension, digital art" ,
    "image_with_transparent_right_half.png" ,
)

 ✔️ Task created with ID: task-xsuhOthvBXLEjddn3ynyiiOR
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!

Autres langues

Package Nodejs