Views
No views yet
1from inference import load_model, predict 2import numpy as np 3 4model = load_model("mnist_cnn.pth") 5 6image = np.zeros((28, 28)) 7prediction = predict(image, model) 8print(prediction) 9