📚 寓言铸造
📄 描述
使用OpenAI的新函数调用和Replicate的API为Stable Diffusion,从单个提示生成图画书。将所有生成的图像和相应的提示存储在Deep Lake中。查看example.pdf
或观看下方视频,了解输出效果。
使用LangChain、Deep Lake和Replicate构建。
:tv: 演示
https://github.com/e-johnstonn/FableForge/assets/30129211/f9523905-342e-4a33-914d-acd13bd168ec
🛠 安装
- 克隆仓库
- 安装requirements.txt
- 在
keys.env
中设置OpenAI和Replicate API密钥 - 下面有更多说明 - 要保存图像和提示,在
keys.env
中设置Activeloop Deep Lake令牌和数据集路径 - 下面有更多说明 - 运行
streamlit run main.py
启动应用!
🧠 Deep Lake设置
在创建这个项目时,我使用Deep Lake将生成的图片和提示存储在云端,因为它可以轻松处理多种数据模态(图像/文本),并在Web界面中显示它们。要自行设置,请访问Deep Lake网站并创建账户。登录后,点击"Train deep learning models",然后"Create dataset",这将指导你获取API令牌和数据集链接。将令牌和数据集路径放入keys.env
文件中即可。
🖼️ Replicate设置
此应用需要Replicate API密钥。要获取一个,请访问Replicate网站并创建账户,然后将API密钥放入keys.env
中。Replicate为新用户提供免费的图像生成服务。
📐架构
改进
- 本演示使用Replicate进行图像生成,因为它易于使用。将其连接到你自己的Stable Diffusion设置(本地或云端)以获得更好的结果。我推荐使用Diffusers和FastAPI的某种组合作为起点。