文章目录
- 效果
- 群成员发送的词条生成sd的prompt
- 中文词条翻译为英文
- 过滤NSFW
- 代码
效果
群成员发送的词条生成sd的prompt
上一篇内容我们已经将sd的文生图API接入了微信群,但是群成员只能通过指定关键词触发sd的调用而生成图片,但我们在实际使用sd时是可以自定义prompt的,从而使得sd生成符合我们预期的图片。
这篇内容我们将实现获取群成员发送的词条,然后将词条传送给sd,从而使得sd按照词条生成图片!
中文词条翻译为英文
由于大多数群成员发送的是中文,而sd的prompt必须是英文,为此我们需要将群成员发送的中文词条翻译为英文。笔者选择了一个可以将中文翻译为英文的python库translate
安装translate
pip install translate
然后我们在上篇内容中的文生图的方法txt2Img
开头进行中文翻译成英文的代码编写