在网页上轻松展示CV、NLP模型,爸爸也可以操作-斯坦福产品-量子比特
“其实我在做机器学习模型。”
“什么模型?拿出来给我看看!”
如果你是一名机器学习(ML)程序员,在向父母、好友和客户介绍自己的职业时,你可能会遇到这种尴尬。
现在有斯坦福大学的Gradio,你真的可以“搬运”自己的ML模型。
只需创建几行代码,就可以创建一个简单的页面,即使是完全不会代码的父母也可以通过拖动操作体验ML模型。
此外,Gradio还可以嵌入Jupyter和Colab笔记本、博客和网站,还可以在研究论文中加入截图进行展示。
用CV、NLP模型简单地
用户可以通过拖动本地图像、粘贴文本、拍照等方式获取输入,在网页的另一侧瞬间出现输出结果。
在计算机上手写数字,实现MNIST识别:
用电脑拍照,检测你现在的心情:
Gradio您认为只能显示计算机视觉任务吗?No!NLP任务也OK:
最近,Gradio项目增加了一个ML模型展示平台GradioHub,供所有开发人员上传自己的模型。根据遗传特征预测,所有前沿研究任务如CT图像重建都可以在这个聚合页面上看到。
使用方法
Gradio安装非常方便,pip代码行即可:
Gradio使用现成代码也很方便,只需加几行代码就可以在网页上显示。这是使用了Inception V3的图像分类的例子:
您也可以提供自己的模型,而不是上述预培训模型,或者使用其他类型的非Keras模型。
通过更改接口的输入和输出参数,可以创建不同的接口。当前支持的输入格式为平板电脑、上传图像、相机、文本,输出结果为标签或文字。
传输门
Gradio网站:
https://Gradio.app
源代码:
https://github.com/Gradio-app/Gradio