網頁輕松展示CV、NLP模型，連你老爸都會操作 | 斯坦福出品

曉查 2019-07-11 13:06:49 來源：量子位

“什么模型？拿出來給大家看看啊！”

曉查發自凹非寺

量子位報道 | 公眾號 QbitAI

“其實我是做機器學習模型的。”

“什么模型？拿出來給大家看看啊！”

如果你是一個機器學習（ML）程序員，向爸媽、親友或者客戶介紹自己的職業時，可能會遇到這樣的尷尬。

現在有了斯坦福大學出品的Gradio，你真的可以把自己的ML模型“搬”出來了。

只需幾行代碼，你就可以創建一個簡單的網頁，即使完全不會代碼的爸媽，也能用拖拽操作體驗你的ML模型。

除此之外，Gradio還可以嵌入Jupyter和Colab筆記本、博客和網站，甚至能把截圖放在研究論文里作為展示。

CV、NLP模型輕松搞定

Gradio允許用戶通過拖拽本地圖像、粘貼文本、拍照等方式獲取輸入，網頁另一側瞬間出現輸出結果。

在電腦上手寫一個數字，實現MNIST識別：

用電腦攝像頭拍一張圖，檢測你現在的情緒：

你以為Gradio只能展示計算機視覺任務？No！NLP任務也OK：

最近，Gradio項目又上線了ML模型展示平臺GradioHub，供所有開發者上傳自己的模型。遺傳特征預測、CT圖像重建這類前沿研究任務都能這個聚合頁面看到。

使用方法

Gradio的安裝非常方便，一行pip代碼即可：

Gradio使用現成代碼也很方便，只需加上幾行代碼就能在網頁展示。這是一個使用Inception V3進行圖像分類的例子：

你也可以提供自己的模型，而不是上面的預訓練模型，或者使用Keras之外其他類型的模型。

更改Interface中的input和output參數可以創建不同的接口。目前Gradio支持的輸入格式有：手寫板、上傳圖像、攝像頭、文本，輸出結果為標簽或文字。

傳送門

Gradio網站：

https://gradio.app

源代碼：

https://github.com/gradio-app/gradio