LLM 课程文档
与他人分享演示
并获得增强的文档体验
开始使用
与他人分享演示
现在你已经构建了一个演示,你可能想要与他人分享它。Gradio 演示可以通过两种方式分享:使用临时分享链接或在 Spaces 上永久托管。
我们稍后将介绍这两种方法。但在你分享演示之前,你可能想先完善它 💅。
完善你的 Gradio 演示:


为了向你的演示添加额外内容,Interface
类支持一些可选参数
title
:你可以为你的演示提供一个标题,它会出现在输入和输出组件的上方。description
:你可以为界面提供一段描述(文本、Markdown 或 HTML 格式),它会出现在输入和输出组件的上方,标题的下方。article
:你还可以编写一篇扩展文章(文本、Markdown 或 HTML 格式)来解释界面。如果提供,它将出现在输入和输出组件的下方。theme
:不喜欢默认颜色?将主题设置为default
、huggingface
、grass
、peach
中的一个。你还可以添加dark-
前缀,例如dark-peach
表示深色主题(或者直接使用dark
表示默认深色主题)。examples
:为了使你的演示更易于使用,你可以为函数提供一些示例输入。这些示例会出现在 UI 组件下方,并可用于填充界面。这些示例应以嵌套列表的形式提供,其中外部列表包含样本,每个内部列表包含与每个输入组件对应的输入。live
:如果你想让你的演示“实时”运行,意味着你的模型在每次输入更改时都会重新运行,你可以设置live=True
。这对于快速模型来说是有意义的(我们将在本节末尾看到一个示例)。使用上面的选项,我们最终得到一个更完整的界面。运行下面的代码,你就可以和 Rick and Morty 聊天了
title = "Ask Rick a Question"
description = """
The bot was trained to answer questions based on Rick and Morty dialogues. Ask Rick anything!
<img src="https://huggingface.co/spaces/course-demos/Rick_and_Morty_QA/resolve/main/rick.png" width=200px>
"""
article = "Check out [the original Rick and Morty Bot](https://huggingface.co/spaces/kingabzpro/Rick_and_Morty_Bot) that this demo is based off of."
gr.Interface(
fn=predict,
inputs="textbox",
outputs="text",
title=title,
description=description,
article=article,
examples=[["What are you doing?"], ["Where should we time travel to?"]],
).launch()
使用上面的选项,我们最终得到一个更完整的界面。试试下面的界面
使用临时链接分享你的演示
现在我们有了一个可用的机器学习模型演示,让我们学习如何轻松分享我们界面的链接。通过在 launch()
方法中设置 share=True
,可以轻松地公开分享界面
gr.Interface(classify_image, "image", "label").launch(share=True)
这将生成一个公开的、可分享的链接,你可以发送给任何人!当你发送此链接时,另一侧的用户可以在他们的浏览器中试用该模型长达 72 小时。由于处理过程发生在你的设备上(只要你的设备保持开启!),你无需担心打包任何依赖项。如果你正在使用 Google Colab 笔记本,则始终会自动创建一个分享链接。它通常看起来像这样:XXXXX.gradio.app。尽管链接是通过 Gradio 链接提供的,但我们只是你本地服务器的代理,并且不存储通过界面发送的任何数据。
但是请记住,这些链接是公开可访问的,这意味着任何人都可以使用你的模型进行预测!因此,请确保不要通过你编写的函数暴露任何敏感信息,或允许在你的设备上发生任何关键更改。如果你设置 share=False
(默认值),则只会创建一个本地链接。
在 Hugging Face Spaces 上托管你的演示
你可以传递给同事的分享链接很酷,但如何永久托管你的演示并使其以自己的“space”形式存在于互联网上呢?
Hugging Face Spaces 提供了在互联网上永久托管你的 Gradio 模型的 инфраструктура,免费!Spaces 允许你创建并推送到(公共或私有)仓库,你的 Gradio 界面代码将以 app.py
文件的形式存在。阅读分步教程 以开始使用,或观看下面的示例视频。
✏️ 让我们应用它!
使用到目前为止章节中学到的知识,让我们创建我们在本章第一节中看到的草图识别演示。让我们为我们的界面添加一些自定义设置,并设置 share=True
以创建一个我们可以传递的公共链接。
我们可以从 class_names.txt 加载标签,并从 pytorch_model.bin 加载预训练的 pytorch 模型。通过单击链接并单击文件预览左上角的下载来下载这些文件。让我们看看下面的代码,了解我们如何使用这些文件来加载我们的模型并创建一个 predict()
函数
from pathlib import Path
import torch
import gradio as gr
from torch import nn
LABELS = Path("class_names.txt").read_text().splitlines()
model = nn.Sequential(
nn.Conv2d(1, 32, 3, padding="same"),
nn.ReLU(),
nn.MaxPool2d(2),
nn.Conv2d(32, 64, 3, padding="same"),
nn.ReLU(),
nn.MaxPool2d(2),
nn.Conv2d(64, 128, 3, padding="same"),
nn.ReLU(),
nn.MaxPool2d(2),
nn.Flatten(),
nn.Linear(1152, 256),
nn.ReLU(),
nn.Linear(256, len(LABELS)),
)
state_dict = torch.load("pytorch_model.bin", map_location="cpu")
model.load_state_dict(state_dict, strict=False)
model.eval()
def predict(im):
x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.0
with torch.no_grad():
out = model(x)
probabilities = torch.nn.functional.softmax(out[0], dim=0)
values, indices = torch.topk(probabilities, 5)
return {LABELS[i]: v.item() for i, v in zip(indices, values)}
现在我们有了 predict()
函数。下一步是定义和启动我们的 gradio 界面
interface = gr.Interface(
predict,
inputs="sketchpad",
outputs="label",
theme="huggingface",
title="Sketch Recognition",
description="Who wants to play Pictionary? Draw a common object like a shovel or a laptop, and the algorithm will guess in real time!",
article="<p style='text-align: center'>Sketch Recognition | Demo Model</p>",
live=True,
)
interface.launch(share=True)
注意 Interface
中的 live=True
参数,这意味着草图演示会在有人在草图板上绘制时进行预测(无需提交按钮!)。
此外,我们还在 launch()
方法中设置了 share=True
参数。这将创建一个公共链接,你可以发送给任何人!当你发送此链接时,另一侧的用户可以试用草图识别模型。重申一下,你也可以将模型托管在 Hugging Face Spaces 上,这就是我们能够嵌入上面演示的方式。
接下来,我们将介绍 Gradio 可以与 Hugging Face 生态系统结合使用的其他方式!
< > 在 GitHub 上更新