Debugging a Federated Learning Project

I have a digit recognition ML project using the Flower framework. I have written all the code but getting errors in my model fit round, can't seem to understand from where the value error is coming. The code is fully done, there's no need to write any further code, all you need to do is debugging