Appreciate the help, harsha_g .
I’m getting the error: Attribute Error: ‘Compose’ object has no attribute ‘unsqueeze’.
Here’s my code for testing my image on the model, any help would be great!
‘’’
Load and Process a Single Image
new_image = Image.open(‘new_image.jpg’)
transformed_image = transforms.Compose([
transforms.ToPILImage(),

transforms.Resize((28, 28)),

transforms.RandomGrayscale(p = 1.0), # Transforms the image to a gray-scale

transforms.ToTensor(),

])
Add a Batch Dimension
transformed_image = transformed_image.unsqueeze(0)
check_image(transformed_image, model)