Problem with pipeline on custom model
I built a custom transformer  for a SequenceClassification task. Everything works fine but when I want to load my model using pipeline, I get wrong results.model_ckpt = 'roberta-base'
tokenizer = AutoTokenizer.from_pretrained(model_ckpt)

labels =['health','business', 'politics', 'entertainment', 'sports', 'tech']
index2cat = {idx:cat for idx, cat in enumerate(labels)}
cat2index = {cat: idx for idx, cat in enumerate(labels)}Here is my custom SequenceClassification modeldef_777×564 77.4 KBTrainer and training argstrain_556×538 57.9 KBUsing pipelineCalling pipeline with the task, model and tokenizer gives the correct results but with the model ID on hub or local directory, I get wrong results. See sample belowpipe_995×425 52.3 KBI have been on this for few days now and  still cannot have a breakthrough at solving it. Thank you