prompt = input(“Image prompt: “)

res = client.images.generate(

model=”dall-e-2″,

prompt=prompt,

n=2,

size=”256×256”

)

print(“Assistant: ” + res.data[0].url)

for image variations:

image_variation_data = client.images.create_variation(

image=open(“./images/img_sample.png”, “rb”),

n=1,

size=”256×256″

)