Bug Report for handwritten-digit-classifier

Bug Report for https://neetcode.io/problems/handwritten-digit-classifier

Please describe the bug below and include any steps to reproduce the bug or screenshots if possible.

The attached solution to the problem doesn't pass the tests.