How could I feed a custom image into this model?
I've been following a course online and one of the exercises was to create a simple image detection model (using MNIST data) to detect written numbers. I've been trying to load a custom image I drew in (128x128 jpg) but I can't seem to figure it out. I'm really close, but I think I'm just confused about what parameters the model takes in. Any help would be appreciated!!

Here is my code

Simply convert your image to an 128x128 numpy array with values between 0 and 1. 

Then:

image = Variable(torch.from_numpy(image))[None, :, :]
classification = model(image)


classification is then a pytorch Variable containing probabilities of belonging to each class. 
