from transformers import pipeline

# 加载模型
generator = pipeline("text-to-image", model="runwayml/stable-diffusion-v1-5")

# 定义提示词
prompt = "    海报设计概念

标题

声扬
放大你的频道影响力

视觉元素

背景：使用渐变色彩，代表音乐的多样性和活力，如蓝色到紫色的渐变。

图像：加入音波图形或音乐符号，象征声音的传播与影响力。

插图：可以添加一些社交媒体图标，表示与平台的连接。

文案

主打口号：
“让音乐为你的频道增添力量！”

服务介绍：

定制专属频道音乐

提升观众参与感

增强品牌识别度

联系方式

二维码：扫描获取更多信息或试听样品。

公司名称和联系方式：清晰展示在底部。

色彩方案

主色调：蓝色、紫色、白色，传达现代感和专业性。

字体选择：选择简洁、现代的无衬线字体，确保易读性。

布局建议

标题居中放置，视觉中心突出。

服务介绍和联系方式在海报底部，确保观众容易找到。  "

# 生成图像
image = generator(prompt, num_inference_steps=50, guidance_scale=7.5)
image[0].save("my_image.png")