Sequences - C4W2 assignment

Hi,
I have finished the assignment and succesfully done all the test and reached the required accuracy.
However, when i submit my assignment, i get this error:
There was a problem grading your submission. Details:
Incompatible shapes: [361,5] vs. [361] [Op:SquaredDifference]

I have reviewed my code over and over and i cannot figure it out. I can’t even know where in my code my error might be. Can anyone help please?
thank you

I solved… it has to do with the NN model architecture. Make sure you look at the model summary, especially the output column.