Image Classification Model from Scratch using PyTorch and Python