Update README.md

README.md

@@ -27,6 +27,8 @@ class ESGify(MPNetPreTrainedModel):
         super().__init__(config)
         # Instantiate Parts of model
         self.mpnet = MPNetModel(config,add_pooling_layer=False)
+        self.id2label =  config.id2label
+        self.label2id =  config.label2id
         self.classifier = torch.nn.Sequential(OrderedDict([('norm',torch.nn.BatchNorm1d(768)),
                                                 ('linear',torch.nn.Linear(768,512)),
                                                 ('act',torch.nn.ReLU()),