Error: Can't convert non-rectangular Python sequence to Tensor while calling neural network a second time

Question

I\'ve defined a Function for my neural network. I run my code on Jupyter Notebook and the code works absolutely fine when I run the function and call the function afterwards. If