批量生成图片
def batch_generate(prompts, output_dir=output):
if not os.path.exists(output_dir):
os.makedirs(output_dir)
for i, prompt in enumerate(prompts):
print(f正在生成第 {i+1} 张图片...)
try:
image_url = generate_image(prompt)
下载图片到本地
import requests
img_data = requests.get(image_url).content
with open(f{output_dir}/image_{i+1}.jpg, wb) as f:
f.write(img_data)
print(f第 {i+1} 张图片已保存!)
except Exception as e:
print(f生成第 {i+1} 张图片时出错:{e})
示例提示词列表
prompts = [
A futuristic cityscape at sunset,
A cute cartoon cat wearing sunglasses,
An abstract painting of a forest,
A realistic portrait of a young woman
]