%20(2).svg)
项目简介
这是一个基于Spring Boot框架的AI聊天机器人界面应用，集成了ChatGLM模型和Stable Diffusion模型，旨在通过智能聊天和图像生成功能，为用户提供交互式体验。
项目的主要特性和功能
- 智能聊天：借助ChatGLM模型实现自然语言处理，可与用户进行智能对话。
- 图像生成：结合Stable Diffusion模型，依据用户输入的文本描述生成对应图像。
- 用户界面：提供友好的Web界面，支持文本输入、提交以及图像展示等功能。
- 状态管理：通过Spring Boot框架实现状态管理，涵盖历史记录、结果列表等。
- 服务器设置：通过Spring Boot启动服务器，支持本地浏览器访问。
安装使用步骤
假设用户已经下载了本项目的源码文件，按以下步骤操作：
1. 复制项目并安装依赖：
shell
cd ChatGLM-ArtAgent
pip install -r requirements.txt
2. 安装NLTK模型及数据：
shell
python
import nltk
nltk.download('stopwords')
nltk.download('punkt')
3. 调整模型接口：
修改utils.py中的API调用地址，以匹配ChatGLM - 6B和Stable Diffusion的实际部署地址。
4. 运行项目：
shell
python art_agent.py
访问http://127.0.0.1:6006即可使用该应用。
硬件与环境需求
- 硬件需求：需要具备一定的计算能力，能够运行ChatGLM和Stable Diffusion模型。
- 环境需求：需要安装Python、NLTK库，以及相关的依赖库。
待办事项
- 完善文档和UI：包括Readme、About us、UI细节等。
- 建立评测体系：构建评测prompt集、评价指标、自动评测脚本等。
- 数据集构建与模型微调：利用艺术评论领域的问答数据集微调GLM - 6B模型。
- 优化Natural Language -> Prompt模块：包括更长的问答链路、补充Prompts的专用模型等。
- 实现类autoGPT链路：并做出创新性优化。
- 图像理解功能：实现上传图像并进行理解。
- 丰富功能：如显示当前prompts、直接与SD交互等。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】