Nevermind, I figured out why. There was a problem with the preprocessing before training and re-trained the whole model again and it worked.
Nevermind, I figured out why. There was a problem with the preprocessing before training and re-trained the whole model again and it worked.