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.

Hello,

I believe the expected output might be incorrect. Even when I copy and paste the solution, my output does not match the expected output.

Have a good day.