使用gradio搭建服务(一)

原创已于 2024-04-11 15:32:01 修改 · 3.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

于 2023-06-07 15:17:30 首次发布

文章展示了如何利用Gradio库创建交互式应用程序，包括处理文本、图像和复选框输入，以及展示文本和图像输出。示例中包含了图像翻转和数据框展示的功能，可用于简单的UI原型设计或模型演示。

使用参考gradio官方教程

三种基本输入：image，text，checkbox
多个输出，gr.outputs实现

import gradio as gr
import numpy as np
def predict(prompt,image,is_rain):
    print('prompt',type(prompt))
    print('image',type(image))
    print('is_rain',type(is_rain))

    image = np.fliplr(image)

    if prompt:
        output1 = prompt.upper()
        output2 = prompt.lower()
    else:
        output1=0
        output2 = 0
    return output1, output2, image

outputs = [
    gr.outputs.Textbox(label="Uppercase"),
    gr.outputs.Textbox(label="Lowercase"),
    gr.outputs.Image(type = 'numpy',label="Flipped Image")

]

demo = gr.Interface(
    fn=predict,
    inputs=[ "text",'image','checkbox'],
    outputs=outputs,
)
demo.launch()

输出

prompt <class 'str'>
image <class 'numpy.ndarray'>
is_rain <class 'bool'>

在这里插入图片描述
图片左边3个框是输入部分，显示的名字为传入参数的名字。右边的三个框是输出部分，名字是output中定义的label

如果要同一局域网的其他设备访问服务，设置最后启动的launch(server_name="0.0.0.0")，访问的时候将127.0.0.1更换成运行设备的ip

显示表格

import gradio as gr
import numpy as np

import gradio as gr
import pandas as pd

def display_table(image):
    data = {'Name': ['Alice', 'Bob', 'Charlie'],
            'Age': [25, 30, 35],
            'City': ['New York', 'London', 'Paris']}
    df = pd.DataFrame(data)
    
    return df


# iface.launch()

if __name__ == "__main__":
    # demo.launch()
    demo = gr.Interface(fn=display_table,inputs=['image'], outputs="dataframe")
    demo.launch(server_name="0.0.0.0") ## ip地址+端口号