文章目录
- 前言
- 一、Gradio是什么?
- 二、使用Gradio构建基于ChatGPT的 Web 应用
- 1. 安装gradio库
- 2. 安装openai库(ChatGPT的python库)
- 3. Web 应用示例(源代码)
- 总结
前言
都说ChatGPT也可以生成图片,好奇的我也尝试包装了一个方法,在本地就可以通过Python访问openai并将结果展现出来,具体怎么做,请看下文。
一、Gradio是什么?
Gradio是一个开源的 Python 库,用于构建机器学习和数据科学演示和 Web 应用。
官网:https://www.gradio.app/
Gradio适用于:
- 演示客户/合作者/用户/学生的机器学习模型。
- 部署通过自动共享链接快速创建模型,并获得模型性能反馈。
- 排除故障使用内置的操作和解释工具,在开发过程中以交互方式处理您的模型。
案例展示:https://huggingface.co/spaces/hysts/insightface-SCRFD
安装包:https://pypi.org/project/gradio/
Github:https://github.com/gradio-app/gradio
二、使用Gradio构建基于ChatGPT的 Web 应用
1. 安装gradio库
pip install gradio -i https://pypi.tuna.tsinghua.edu.cn/simple
2. 安装openai库(ChatGPT的python库)
pip install openai -i https://pypi.tuna.tsinghua.edu.cn/simple
请提前申请好账号,并生成密钥。
3. Web 应用示例(源代码)
注:openai.api_key 需更换为自己的密钥
import openai
import gradio as gr
from gradio.components import Textbox,Image
openai.api_key = "sk-xxx" # 更换为自己的密钥
def ChatGPT_AI_DrawBot(input):
if input:
response = openai.Image.create(
prompt=input,
n=1,
size="1024x1024"
)
image_url = response['data'][0]['url']
return image_url
inputs = Textbox(lines=7, label="请输入你的问题")
outputs = Image(label="ChatGPT AI结果")
gr.Interface(fn=ChatGPT_AI_DrawBot, inputs=inputs, outputs=outputs, title="ChatGPT AI助理",
description="我是您的AI助理,输入您的绘图关键字,我将为您进行关键字的AI绘图",
theme=gr.themes.Default()).launch(share=True)
运行后出现以下提示即为成功。
按Ctrl并鼠标点击http://127.0.0.1:7860 即可打开浏览器访问该网址。
给ChatGPT写入一段话:
在一片巨大的森林里,数十棵巨大的百年橡树,金色的秋天,浩瀚的晨光,穿过树枝的天空,一棵长着翡翠叶子的石头树,地面上有巨大的水晶和钻石,所有这些都有多种颜色,淡蓝色的雾,金色的部分,细节,真实感,照片
Submit后,ChatGPT运算后会将生成的图片展示在右侧。点击图片右上角即可下载该图片(直接下载会保存为image.txt,修改扩展名为jpg后即可显示为图片)。如下所示:
最终生成的1024*1024像素图片:
总结
这里是ChatGPT 3.5引擎生成的图片,仅作一个展示,抛砖引玉。期待大家更好的作品!