Gradio中Button用法及事件監(jiān)聽器click方法使用

瞎想亂記

事情是這樣的：入職時(shí)面試的是Java，簡歷中寫了會(huì)python，剛好最近有個(gè)小項(xiàng)目需要用Python實(shí)現(xiàn)，老板就將這個(gè)項(xiàng)目交給了我，我…

項(xiàng)目中還真遇到了好幾個(gè)坑，分享一下

功能介紹

功能很簡單，就是一個(gè)添加一個(gè)按鈕，并給這個(gè)按鈕添加一個(gè)監(jiān)聽器，實(shí)現(xiàn)click方法（我一個(gè)后端搞起了前端，以后請(qǐng)叫我全棧）。

相關(guān)代碼

Gradio的基礎(chǔ)不再介紹了，不會(huì)的可以去先看看基礎(chǔ)

按鈕的代碼如下：

import gradio as gr
submit_button = gr.Button('提交')

以上是定義了一個(gè)名為“提交”的按鈕，我想實(shí)現(xiàn)通過點(diǎn)擊此按鈕實(shí)現(xiàn)某個(gè)方法，輸出一行文字，方法的入?yún)⒂腥齻€(gè)參數(shù)

具體代碼如下：

image = gr.Image(label="圖片", sources=["upload"], type="filepath")
video = gr.Video(label="視頻", sources=["upload"], format='mp4')
audio = gr.Audio(label="音頻", sources=["upload"])
output_video = gr.Text()
submit_button.click(fun_train_video, inputs=[image, video, audio], outputs=output_video)

稍微解釋下這行代碼，click是按鈕的監(jiān)聽器，當(dāng)點(diǎn)擊“提交”按鈕后，會(huì)運(yùn)行函數(shù)fun_train_video，這個(gè)函數(shù)的入?yún)⒂腥齻€(gè)，

分別是：image，video，audio，函數(shù)的輸出是output_video。

這樣的代碼pycharm并沒有報(bào)錯(cuò)，但是在運(yùn)行時(shí)會(huì)出下下面的錯(cuò)誤

“inputs”: [block._id for block in self.inputs],
^^^^^^^^^
AttributeError: ‘tuple’ object has no attribute ‘_id’

解決方法

查了很多資料，包括chatGPT，但是并沒有得到理想的答案，問題遲遲沒有解決，眼看著項(xiàng)目要交付，著急啊，后來想起了我的研究生室友，人稱憨憨，上學(xué)時(shí)寫的一手好python，果斷請(qǐng)教，最后得到解決。

其實(shí)這個(gè)問題的根源是這樣的：在監(jiān)聽器click方法中，inputs參數(shù)需要傳入一個(gè)包含輸入組件的列表，但是inputs=[image, video, audio]這種寫法是包含元組的列表，這個(gè)元組包含了輸入組件和它的標(biāo)簽，但不是組件本身。

當(dāng)gradio嘗試獲取輸入組件id的時(shí)候，會(huì)從元組中獲取_id屬性，但是元組中沒有這個(gè)屬性，就會(huì)拋出異常。

修改代碼如下

submit_button.click(fun_train_video, inputs=[image[0], video, audio], outputs=output_video)

那就是在第一個(gè)參數(shù)后邊加一個(gè)序號(hào)，這樣gradio就可以找到組件的id了，挺神奇的，更深層的邏輯還有待去了解，問題總算解決了。

總結(jié)

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評(píng)論