Python之Gradio简单使用

news2025/2/25 19:55:02

安装Gradio

pip install gradio

示例用法

使用Gradio创建了一个简单的应用程序，将输入文本进行反转并返回结果。在启动应用程序时，您还添加了共享选项和身份验证。

import gradio as gr
def reverse(text):
    return text[::-1]
demo = gr.Interface(reverse, "text", "text")
demo.launch(share=True, auth=("username", "password"))

在这个示例中，reverse函数接受一个文本输入并返回反转后的文本。然后，您使用gr.Interface类创建了一个接口对象demo，指定输入类型为"text"（文本输入）和输出类型为"text"（文本输出），并将reverse函数作为处理函数传递给gr.Interface。

最后，使用demo.launch()方法来启动Gradio应用程序。您还传递了share=True参数，允许其他人通过生成的URL来访问您的应用程序，并且添加了身份验证信息auth=("username", "password")来限制访问权限。

运行该代码后，您将在命令行中看到一个本地的URL地址，类似于http://127.0.0.1:7860/。将此URL复制到浏览器中，您将能够访问该应用程序，并在输入框中输入文本，看到反转后的结果。

运行界面：
访问http://127.0.0.1:7860/会显示登录界面。
在这里插入图片描述
输入账号密码后的界面：

应用界面

Gradio提供了两种主要的组件来创建交互式界面：gr.Interface和gr.Blocks。它们都可以用于构建Web界面，但在功能和使用方式上有所不同。

1. gr.Interface

是Gradio的主要接口类，用于创建简单而强大的交互式应用程序。它使您能够定义输入和输出的类型，并使用处理函数来处理输入并生成输出。您可以指定不同的输入和输出类型，包括文本、图像、音频、视频等。

代码实例：
这段代码使用Gradio创建了一个简单的图像分类器应用程序。它接受一张图像作为输入，并返回一个包含分类结果的字典。

import gradio as gr

def image_classifier(inp):
    return {'cat': 0.3, 'dog': 0.7}

demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
demo.launch()

运行界面：
在这里插入图片描述
在实例中，image_classifier函数是处理函数，它接受一个图像输入并返回一个包含分类结果的字典。在这个示例中，为了简化，我们返回了一个固定的结果字典 {'cat': 0.3, 'dog': 0.7}，其中表示该图像有 30% 的概率是猫，70% 的概率是狗。

使用gr.Interface类创建了一个接口对象 demo，指定输入类型为 “image”（图像输入）和输出类型为 “label”（标签输出）。在这个示例中，我们只使用了默认的图像和标签组件，因此不需要显示地定义输入和输出界面。

最后，使用 demo.launch() 方法启动Gradio应用程序，并在本地的 Web 服务器上运行它。

运行该代码后，您将在命令行中看到一个本地的 URL 地址，类似于 http://127.0.0.1:7860/。将此 URL 复制到浏览器中，您将能够访问该应用程序，并上传图像进行分类。

请注意，为了使图像分类器真正有效，您需要将 image_classifier 函数实现为一个真实的模型，该模型将根据输入图像进行实际的分类预测。

2. gr.Blocks

是Gradio的另一个功能，它提供了一种可视化方式来组合和重用交互式组件。通过使用gr.Blocks，您可以创建复杂的界面布局，并将输入、输出和其他组件组合在一起。每个块都有自己的输入和输出，可以连接到其他块，以构建数据流和处理流程。

代码实例：
使用Gradio的gr.Blocks创建的交互式界面示例。它允许用户输入姓名，并在点击按钮后显示欢迎消息。

import gradio as gr
def update(name):
    return f"Welcome to Gradio, {name}!"

with gr.Blocks() as demo:
    gr.Markdown("Start typing below and then click **Run** to see the output.")
    with gr.Row():
        inp = gr.Textbox(placeholder="What is your name?")
        out = gr.Textbox()
    btn = gr.Button("Run")
    btn.click(fn=update, inputs=inp, outputs=out)

demo.launch()