请你生成一段可以根据提示生成二次元图片的python代码
时间: 2024-06-09 15:10:00 浏览: 109
使用Python GAN 生成二次元头像【100011884】
5星 · 资源好评率100%
好的,请看以下代码:
```python
import requests
from io import BytesIO
from PIL import Image
# 输入提示
prompt = "生成一张可爱的二次元女孩图片"
# 发送请求
response = requests.post(
"https://api.openai.com/v1/images/generations",
headers={
"Content-Type": "application/json",
"Authorization": "Bearer API_KEY" # 替换为你的API Key
},
json={
"model": "image-alpha-001", # 模型名称
"prompt": prompt,
"num_images": 1, # 生成的图片数量
"size": "1024x1024", # 图像大小
"response_format": "url"
}
)
# 获取生成的图片URL
image_url = response.json()["data"][0]["url"]
# 下载并显示生成的图片
response = requests.get(image_url)
image = Image.open(BytesIO(response.content))
image.show()
```
请注意,这里调用了OpenAI的API,需要替换代码中的API Key。同时,这里使用的是OpenAI提供的预训练模型,可能需要额外的费用。如果你想使用其他模型或自己的模型,则需要进行相应的修改。
阅读全文