ValueError: shape mismatch: value array of shape (5923,784) could not be broadcast to indexing result of shape (6000,784)

- getting this error trying to run the nerual network

<img width="1257" alt="image" src="https://user-images.githubusercontent.com/50878033/155981010-2462bc56-e865-4915-ac75-1a56caf52f5c.png">
