abdurafay19
/

Digit-Classifier

Image Classification

digit-recognition

Model card Files Files and versions

abdurafay19 commited on Mar 11

Commit

541d601

·

verified ·

1 Parent(s): ddef3d5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -91,7 +91,7 @@ from model import Model  # your model definition
 # Load model
 model = Model()
-model.load_state_dict(torch.load("mnist_best.pth"))
 model.eval()
 # Preprocess image

 # Load model
 model = Model()
+model.load_state_dict(torch.load("model.pt"))
 model.eval()
 # Preprocess image