Hub

TF-Keras (legado)

tf-keras é o nome dado à versão 2.x do Keras. Agora é hospedado como um repositório separado no GitHub aqui. Embora seja um framework legado, ainda há mais de 4,5 mil modelos hospedados no Hub. Esses modelos podem ser carregados usando a biblioteca huggingface_hub. Você deve ter tf-keras ou keras<3.x instalado na sua máquina.

Se você tem interesse no suporte ao Keras 3.x, confira este guia.

Depois de instalado, basta usar o método from_pretrained_keras para carregar um modelo do Hub. Leia mais sobre from_pretrained_keras aqui.

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))]}!')

<CopyLLMTxtMenu containerStyle="float: inline-end; margin-inline-start: 10px; display: inline-flex; position: relative; z-index: 10;"></CopyLLMTxtMenu>

# The image is a sunflower!

Você também pode hospedar seu modelo tf-keras no Hub. No entanto, tenha em mente que tf-keras é um framework legado. Para alcançar o maior número possível de usuários, recomendamos criar seu modelo usando Keras 3.x e compartilhá-lo nativamente, conforme descrito acima. Para mais detalhes sobre o upload de modelos tf-keras, consulte a documentação de 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,
)

Recursos adicionais

• Repositório no GitHub

• Post no blog Colocando o Keras no 🤗 Hub para Treinamento Colaborativo e Reprodutibilidade (abril de 2022)