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)
|