ylecun/mnist
Viewer • Updated • 70k • 101k • 258
How to use sureshkrishnan/huggingface_mnist_upload with Keras:
# Available backend options are: "jax", "torch", "tensorflow".
import os
os.environ["KERAS_BACKEND"] = "jax"
import keras
model = keras.saving.load_model("hf://sureshkrishnan/huggingface_mnist_upload")
This model is trained on the MNIST dataset to recognize handwritten digits (0–9). It uses a simple CNN architecture and is saved in Keras format.
Load the model using TensorFlow/Keras:
from tensorflow import keras
model = keras.models.load_model("mnist_model.keras")
Uploaded by Suresh Krishnan for Hackathon.