deepshah23 commited on
Commit
c545bcd
·
verified ·
1 Parent(s): 59cca60

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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("e_mnist_digit_blank_cnn_ts_v1.pt")
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)