Hub

TF-Keras (legacy)

tf-keras adalah nama yang diberikan untuk versi Keras 2.x. Saat ini, framework tersebut dihosting sebagai repositori GitHub terpisah di sini. Meskipun merupakan framework legacy, masih ada 4.5k+ model yang dihosting di Hub. Model-model ini dapat dimuat menggunakan pustaka huggingface_hub. Anda harus memiliki tf-keras atau keras<3.x yang terinstal di mesin Anda.

Jika Anda tertarik dengan dukungan Keras 3.x, lihat panduan ini.

Setelah terinstal, Anda hanya perlu menggunakan metode from_pretrained_keras untuk memuat model dari Hub. Baca lebih lanjut tentang from_pretrained_keras di sini.

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!

Anda juga dapat meng-host model tf-keras Anda di Hub. Namun, perlu diingat bahwa tf-keras adalah framework legacy. Untuk menjangkau jumlah pengguna sebanyak mungkin, kami merekomendasikan untuk membuat model Anda menggunakan Keras 3.x dan membagikannya secara native seperti yang dijelaskan di atas. Untuk detail lebih lanjut tentang mengunggah model tf-keras, lihat dokumentasi 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,
)

Sumber daya tambahan

• repo GitHub

• Artikel blog Menerapkan Keras di 🤗 Hub untuk Pelatihan Kolaboratif dan Reproduktibilitas (April 2022)