使用 Gradio 2.0 结合 Hugging Face 模型

发布于 2021 年 5 月 25 日
在 GitHub 上更新
交叉发布自 Gradio 博客

Hugging Face 模型中心拥有超过 10,000 个用户提交的机器学习模型。您会找到各种自然语言处理模型,例如芬兰语和英语之间的翻译模型,或中文语音识别模型。最近,该中心甚至扩展到包括图像分类和音频处理模型。

Hugging Face 一直致力于使模型易于访问和使用。transformers 库使得只需几行代码即可加载模型。加载模型后,它可以通过编程方式用于对新数据进行预测。但使用机器学习模型的不只是程序员!在机器学习中,越来越常见的情况是向跨学科团队演示模型或让非程序员使用模型(以帮助发现偏差、故障点等)。

Gradio 库让机器学习开发者可以非常轻松地从机器学习模型创建演示和 GUI,并像分享 Google 文档链接一样轻松地与协作者免费分享它们。现在,我们很高兴地分享 Gradio 2.0 库让您只需 1 行代码即可加载和使用几乎任何带 GUI 的 Hugging Face 模型。这是一个例子

GIF of Gradio 2.0

默认情况下,这使用 HuggingFace 托管的推理 API(您可以提供自己的 API 密钥或在没有 API 密钥的情况下使用公共访问),或者如果您愿意,也可以运行 pip install transformers 并在本地运行模型计算。

您想自定义演示吗?您可以通过传入自己的参数来覆盖 Interface 类的任何默认参数

GIF of Gradio 2.0

等等,还有更多!模型中心已有 10,000 个模型,我们不将模型视为独立的程序代码,而是可以组合和混合以创建更复杂应用程序和演示的乐高积木。

例如,Gradio 允许您并行加载多个模型(想象一下您想比较 Hugging Face 的 4 个不同文本生成模型,看看哪个最适合您的用例)

GIF of Gradio 2.0

或者将您的模型串联起来。这使得构建由多个机器学习模型组成的复杂应用程序变得容易。例如,在这里我们只需 3 行代码即可构建一个用于翻译和总结芬兰新闻文章的应用程序

GIF of Gradio 2.0

您甚至可以并行地将多个模型串联起来(我们让您自己尝试!)。要尝试这些,只需安装 Gradio (pip install gradio) 并选择您想尝试的 Hugging Face 模型。开始使用 Gradio 和 Hugging Face 进行构建吧 🧱⛏️

社区

注册登录以发表评论