Update README.md
Browse files
README.md
CHANGED
@@ -120,7 +120,7 @@ All 5,000 blank images were correctly classified.
|
|
120 |
import torch
|
121 |
|
122 |
# Load TorchScript model
|
123 |
-
model = torch.jit.load("
|
124 |
model.eval()
|
125 |
|
126 |
# Dummy input (1 image, 1 channel, 28x28)
|
|
|
120 |
import torch
|
121 |
|
122 |
# Load TorchScript model
|
123 |
+
model = torch.jit.load("mnist_emnist_blank_cnn_v1.pt")
|
124 |
model.eval()
|
125 |
|
126 |
# Dummy input (1 image, 1 channel, 28x28)
|