hi author, how can we use this model for multiclass segmentation. where to look in the code. please tell me