快捷導(dǎo)航

Gradio機(jī)器學(xué)習(xí)模型快速部署工具quickstart

更新時(shí)間：2023年04月03日 11:01:04 作者：Livingbody

這篇文章主要為大家介紹了Gradio機(jī)器學(xué)習(xí)模型快速部署工具quickstart，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

引言

書接上回 Gradio機(jī)器學(xué)習(xí)模型快速部署工具【quickstart】翻譯，講到多輸入輸出，其實(shí)很簡單，就是把多個(gè)組件包裝到列表，inputs和outputs對應(yīng)的就是2個(gè)列表，輸入輸出列表，僅此而已。

1.圖像示例

Gradio 支持多種類型的組件，例如Image, DataFrame, Video, 或Label. 讓我們嘗試一個(gè)圖像到圖像的功能來感受一下這些！

import numpy as np
import gradio as gr
def sepia(input_img):
    sepia_filter = np.array([
        [0.393, 0.769, 0.189], 
        [0.349, 0.686, 0.168], 
        [0.272, 0.534, 0.131]
    ])
    sepia_img = input_img.dot(sepia_filter.T)
    sepia_img /= sepia_img.max()
    return sepia_img
demo = gr.Interface(sepia, gr.Image(shape=(200, 200)), "image")
demo.launch()

使用該Image組件作為輸入時(shí)，您的函數(shù)將接收一個(gè)形狀為的 NumPy 數(shù)組(width, height, 3)，其中最后一個(gè)維度表示 RGB 值。我們還將以 NumPy 數(shù)組的形式返回圖像。

您還可以使用關(guān)鍵字參數(shù)設(shè)置組件使用的數(shù)據(jù)類型type=。例如，如果您希望您的函數(shù)采用圖像的文件路徑而不是 NumPy 數(shù)組，則輸入Image組件可以寫為：

gr.Image(type="filepath", shape=...)

另請注意，我們的輸入Image組件帶有一個(gè)編輯按鈕??，它允許裁剪和放大圖像。以這種方式處理圖像有助于揭示機(jī)器學(xué)習(xí)模型中的偏差或隱藏缺陷！

您可以在Gradio 文檔中閱讀更多關(guān)于許多組件以及如何使用它們的信息。

2.塊：更多的靈活性和控制

Gradio 提供了兩個(gè)類來構(gòu)建應(yīng)用程序：

1. Interface，它為創(chuàng)建我們迄今為止一直在討論的演示提供了高級抽象。

2. Blocks，一種用于設(shè)計(jì)具有更靈活布局和數(shù)據(jù)流的網(wǎng)絡(luò)應(yīng)用程序的低級 API。Blocks 允許你做一些事情，比如以多個(gè)數(shù)據(jù)流和演示為特色，控制組件在頁面上出現(xiàn)的位置，處理復(fù)雜的數(shù)據(jù)流（例如，輸出可以作為其他功能的輸入），以及根據(jù)用戶交互更新組件的屬性/可見性——仍然全部在 Python 中。如果您需要這種可定制性，請?jiān)囋?code>Blocks吧！

讓我們看一個(gè)簡單的例子。請注意此處的 API 與Interface.

import gradio as gr
def greet(name):
    return "Hello " + name + "!"
with gr.Blocks() as demo:
    name = gr.Textbox(label="Name")
    output = gr.Textbox(label="Output Box")
    greet_btn = gr.Button("Greet")
    greet_btn.click(fn=greet, inputs=name, outputs=output)
demo.launch()

注意事項(xiàng)：

Blocks是用一個(gè)with子句制作的，在這個(gè)子句中創(chuàng)建的任何組件都會自動(dòng)添加到應(yīng)用程序中。
組件按照它們創(chuàng)建的順序垂直顯示在應(yīng)用程序中。（稍后我們將介紹自定義布局?。?/li>
創(chuàng)建了A Button，然后將click事件偵聽器添加到此按鈕。這個(gè) API 應(yīng)該看起來很熟悉！與一樣Interface，該click方法采用 Python 函數(shù)、輸入組件和輸出組件。

3.更復(fù)雜的 Blocks

這是一個(gè)應(yīng)用程序，可讓您體驗(yàn)Blocks:

import numpy as np
import gradio as gr
# 1.文字翻轉(zhuǎn)
def flip_text(x):
    return x[::-1]
# 2.圖像翻轉(zhuǎn)
def flip_image(x):
    return np.fliplr(x)
with gr.Blocks() as demo:
    # 3.添加markdown組件
    gr.Markdown("Flip text or image files using this demo.")
    # 4.添加Tab
    with gr.Tab("Flip Text"):
        text_input = gr.Textbox()
        text_output = gr.Textbox()
        text_button = gr.Button("Flip")
    with gr.Tab("Flip Image"):
        with gr.Row():
            image_input = gr.Image()
            image_output = gr.Image()
        image_button = gr.Button("Flip")
    # 5.添加Accordion
    with gr.Accordion("Open for More!"):
        gr.Markdown("Look at me...")
    # 6.添加2個(gè)按鈕
    text_button.click(flip_text, inputs=text_input, outputs=text_output)
    image_button.click(flip_image, inputs=image_input, outputs=image_output)
demo.launch()