I cannot call .predict with and I don\'t understand the reason why. It gives me the error: Dimension mismatch. I searched everywhere but I cannot find a solution to my probl