README.md

metadata

tags:
  - image-classification
  - pytorch
  - mnist
license: apache-2.0
library_name: pytorch
pipeline_tag: image-classification

MNIST Digit Classifier

A convolutional neural network trained on MNIST to classify digits 0-9.

Usage

from src.model import DigitClassifier
import torch

model = DigitClassifier()
model.load_state_dict(torch.load("model_weights.pth"))
model.eval()

# Preprocessing (same as training):
transform = transforms.Compose([
    transforms.Resize((28, 28)),
    transforms.Grayscale(),
    transforms.ToTensor(),
    transforms.Normalize((0.1307,), (0.3081,))
])