admin 管理员组

文章数量: 887021

文章目录

  • 前言
  • 一、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/project/gradio/
Github:https://github/gradio-app/gradio

二、使用Gradio构建基于ChatGPT的 Web 应用

1. 安装gradio库

pip install gradio -i https://pypi.tuna.tsinghua.edu/simple

2. 安装openai库(ChatGPT的python库)

pip install openai -i https://pypi.tuna.tsinghua.edu/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引擎生成的图片,仅作一个展示,抛砖引玉。期待大家更好的作品!

本文标签: 小技巧 源码 Gradio python WEB