Hub 文档

TF-Keras(遗留)

Hugging Face's logo
加入 Hugging Face 社区

并获取增强文档体验

开始使用

TF-Keras(旧版)

tf-keras 是 Keras 2.x 版本的名称。现在它作为单独的 GitHub 仓库托管 此处。虽然它是一个旧版框架,但 Hub 上仍然托管着 4500 多个模型。可以使用 huggingface_hub 库加载这些模型。您**必须**在您的机器上安装 tf-keraskeras<3.x

如果您有兴趣了解 Keras 3.x 支持,请查看 本指南

安装完成后,您只需使用 from_pretrained_keras 方法即可从 Hub 加载模型。详细了解 from_pretrained_keras 此处

from huggingface_hub import from_pretrained_keras

model = from_pretrained_keras("keras-io/mobile-vit-xxs")
prediction = model.predict(image)
prediction = tf.squeeze(tf.round(prediction))
print(f'The image is a {classes[(np.argmax(prediction))]}!')

# The image is a sunflower!

您也可以将您的 tf-keras 模型托管在 Hub 上。但是,请记住 tf-keras 是一个旧版框架。为了让更多用户使用,我们建议您使用 Keras 3.x 创建模型,并像上面描述的那样本地共享它。有关上传 tf-keras 模型的更多详细信息,请查看 push_to_hub_keras 文档

from huggingface_hub import push_to_hub_keras

push_to_hub_keras(model,
    "your-username/your-model-name",
    "your-tensorboard-log-directory",
    tags = ["object-detection", "some_other_tag"],
    **model_save_kwargs,
)

其他资源

< > GitHub 上的更新