Using Keras at Hugging Face

keras is an open-source machine learning library that uses a consistent and simple API to build models leveraging TensorFlow and its ecosystem.

Exploring Keras in the Hub

You can find over 200 keras models by filtering at the left of the models page.

All models on the Hub come up with useful feature:

An automatically generated model card with a description, a plot of the model, and more.
Metadata tags that help for discoverability and contain information such as license.
If provided by the model owner, TensorBoard logs are hosted on the Keras repositories.

Using existing models

The huggingface_hub library is a lightweight Python client with utility functions to download models from the Hub.

pip install huggingface_hub["tensorflow"]

Once you have the library installed, you just need to use the from_pretrained_keras method. Read more about from_pretrained_keras here.

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!

If you want to see how to load a specific model, you can click Use in keras and you will be given a working snippet that you can load it!

Sharing your models

You can share your keras models by using the push_to_hub_keras method. This will generate a model card that includes your model’s hyperparameters, plot of your model and couple of sections related to the usage purpose of your model, model biases and limitations about putting the model in production. This saves the metrics of your model in a JSON file as well. Read more about push_to_hub_keras here.

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,
)

The repository will host your TensorBoard traces like below.

Additional resources

Keras Developer Guides.
Keras examples.
Keras examples on 🤗 Hub.
Keras learning resources
For more capabilities of the Keras integration, check out Putting Keras on 🤗 Hub for Collaborative Training and Reproducibility tutorial.