当前位置:首页 » 《休闲阅读》 » 正文

Gradio,快速构建AI算法的交互式web界面

8 人参与  2024年09月28日 14:01  分类 : 《休闲阅读》  评论

点击全文阅读


文章目录

1、前言2、安装3、组件介绍3.1 输入输出组件:3.2 控制组件:3.3 布局组件: 4、演示demo4.1 gr.interface4.2 gr.Blocks

1、前言

Gradio是一个开源的python库,像大多数依赖库一样,可直接pip安装导包使用。它主要做的一件事,是帮助你快速创建一个用户界面,以便向他人演示你的AI模型,尤其是一些对话的大语言模型。另外,Gradio主要是易用,上手简单,快速创建,想必可以省去不少麻烦。

2、安装

Gradio要求python版本在3.7以上(包含),直接使用pip安装即可,建议使用镜像源。

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

3、组件介绍

导包 import gradio as gr,这里用gr代替gradio

3.1 输入输出组件:

gr.Texbos:文本框gr.Image:图像gr.Number:数字gr.Files:文件gr.Dropdown:下拉选项

3.2 控制组件:

gr.Button:按钮gr.Slider:滑块gr.Checkbox:复选框gr.Switch:开关

3.3 布局组件:

gr.Tab:标签页gr.Row:行布局gr.Column:列布局

4、演示demo

Gradio提供了两种方式创建应用界面
1、gr.interface:定义输入和输出函数,将它们与界面组件关联,适用于简单场景应用
2、gr.Blocks:可以使用不同的布局块来组织界面组件,更加适用于定制化场景应用

4.1 gr.interface

gr.Interface封装接口的初始化参数:

fn:要处理的函数inputs:输入的组件类型outputs:输出的组件类型

演示demo,输入一张图像,输出该图像的大小

import gradio as gr#处理函数def greet(img):    shape = img.shape    return shape#定义输入输出组件input = gr.Image(label='请输入一张彩色图')output = gr.Text(label='展示彩色图像的大小')demo = gr.Interface(    fn = greet,    inputs = input,    outputs = output)demo.launch(share=True)

成功执行上述代码之后,在终端会输出web界面链接

在这里插入图片描述

可直接点击链接打开界面,定义的输入组件(图像)以及Label,输出组件(文本框)以及Label(用作提示),左下角的clear(清除) 和 submit (提交)组件是自带的。

在这里插入图片描述

测试一下,上传或拖拽一张图像,点击submit,可看到右侧文本框中显示图像大小

在这里插入图片描述

在这里,输入组件(图像)默认是对图像进行了读操作的,因为上面定义的处理函数,直接对函数输入进行shape了!

4.2 gr.Blocks

gr.Block方式,还是用刚才的demo例子

import gradio as gr#处理函数def greet(img):    shape = img.shape    return shapewith gr.Blocks() as demo:    input = gr.Image(label='请输入一张彩色图')    output = gr.Text(label='展示彩色图像的大小')    #创建控制组件按钮作为提交    submit_btn = gr.Button('submit')    #使用click方法为按钮添加回调函数    submit_btn.click(fn=greet,inputs=input,outputs=output)demo.launch(share=True)

链接进去后是这样的,布局不太一样,测试也是上传图片,点击submit输出结果。

在这里插入图片描述
仍是刚才的demo,不过这里稍加修改下,以行布局形式添加clear按钮

import gradio as gr#处理函数,分两种情况def greet(img):    if img is None:        return 0    else:        shape = img.shape        return shapewith gr.Blocks() as demo:    input = gr.Image(label='请输入一张彩色图')    output = gr.Text(label='展示彩色图像的大小')    #布局组件,以行分布显示submit和clear按钮    with gr.Row():        submit_btn = gr.Button("submit")        clear_btn = gr.Button("clear")        submit_btn.click(fn=greet,inputs=input,outputs=output)        #clear按钮的回调函数输入永远为None        clear_btn.click(fn=greet,inputs=None,outputs=output)demo.launch(share=True)

这里点击clear按钮,输出永远是0,因为这里的回调函数输入永远为None,输出为0

在这里插入图片描述
玩过其他工具的各种组件,相信对gradio工具的组件也会随意组合玩耍滴。


点击全文阅读


本文链接:http://zhangshiyu.com/post/165789.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

关于我们 | 我要投稿 | 免责申明

Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1