用Openai的dall-e2生成图片的代码

使用DALL-E 2生成图片需要先安装OpenAI API和相关的Python库。以下是生成图片的基本代码:

import openai
import requests
from requests.structures import CaseInsensitiveDict

import json

openai.api_key = "YOUR_API_KEY"

image_prompt = "a portrait of a cat"
openai_model = "image-alpha-001"
image_size = 256

def generate_image(prompt, model, size):
    
    headers = CaseInsensitiveDict()
    headers["Content-Type"] = "application/json"
    headers["Authorization"] = f"Bearer {openai.api_key}"

    data = """
    {
        """
    data += f'"model": "{model}",'
    data += f'"prompt": "{prompt}",'
    data += f'"num_images":1,'
    data += f'"size":"{size}"'
    data += """
    }
    """

    resp = requests.post("https://api.openai.com/v1/images/generations", headers=headers, data=data)

    if resp.status_code != 200:
        raise ValueError("Failed to generate image")

    response_text = json.loads(resp.text)
    image_url = response_text['data'][0]['url']

    return image_url

image_url = generate_image(image_prompt, openai_model, image_size)
print("Image URL:", image_url)

这里使用了OpenAI API的“images”功能,通过post请求将生成图片的参数传递给API,获取url链接后,你可以将生成图片的代码进一步完善,实现更多自定义参数的功能,例如生成多张图片、调整图片大小、添加图像元素等。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

技术老张

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值