Gradio：让你的AI算法可视化！

Gradio使用教程（一）

这篇文章主要用于记录在部署ChatGLM2-6B时，遇到的Gradio常见用法。Gradio version=3.48
Gradio官方教程

Gradio的安装

pip install gradio

一个简单的start_demo

执行以下代码，就能得到一个打招呼的网页

import gradio as gr

def greet(name):
    return 'hello,{}'.format(name)
    
demo = gr.Interface(fn=greet, inputs=gr.Textbox(), outputs="label")
demo.launch()

Interface用法

如上文提到的，Gradio.Interface()提供了用python代码来实现web ui界面的功能。他的语法格式如下：

gradio.Interface(fn,inputs,outputs,...)

其中，fn通常是一个机器学习的预测函数，要求他能返回一个值，它接受inputs作为参数，outputs是它返回的值

gradio.Interface.launch(···)

这个语句可以将网页发送至web服务器，它提供了server_name和server_port参数来指定服务器ip和转发端口，默认ip是127.0.0.1，和7860
例如：

demo.launch(
server_name = '10.216.23.34',
server_prot = 5000)

我们就会得到一个地址为http://10.216.23.34:5000/的网页