Hi, I read your code quickly and I found out that this code just works for sequence classification models. is it true?