Update configuration_ced.py

configuration_ced.py

@@ -14,7 +14,7 @@
 # limitations under the License.
 """ CED model configuration"""
 
-
+import csv
 from transformers import PretrainedConfig
 from transformers.utils import logging
 from transformers.utils.hub import cached_file
@@ -127,10 +127,14 @@ class CedConfig(PretrainedConfig):
 
         if self.outputdim == 527:
             with open(cached_file("topel/ConvNeXt-Tiny-AT", "class_labels_indices.csv"), "r") as f:
-                self.id2label = {
-                    int(line.split(",", maxsplit=3)[0]): line.split(",", maxsplit=3)[2].replace('"', "").strip("\n")
-                    for line in f.readlines()[1:]
-                }
+                 reader = csv.reader(f)                                                                                                                                                                 
+                 next(reader)  # skip header                                                                                                                                                            
+                 self.id2label = {}                                                                                                                                                                     
+                 for row in reader:                                                                                                                                                                     
+                     idx = int(row[0])                                                                                                                                                                  
+                     label = row[2]                                                                                                                                                                     
+                     if label not in self.id2label.values():                                                                                                                                            
+                         self.id2label[idx] = label
             self.label2id = {v: k for k, v in self.id2label.items()}
         else:
             self.id2label = None