How do I change the classification head of a model?

I guess you’re right. I’m still looking for a solution HERE