Upload engine.py

Fixed name errors

FcgEngine/engine.py

@@ -59,7 +59,8 @@ class PredictorCls(object):
         self.model = AutoModelForImageClassification.from_pretrained(model_path, trust_remote_code=True)
         self.model.to(device)
         config = AutoConfig.from_pretrained(model_path, trust_remote_code=True)
-        self.ids = list(config.cls_name.keys())
+        self.name_list = list(config.cls_name.keys())
+        self.id_list = list(config.cls_name.values())
         self.device = device
 
     def __call__(self, spectra):
@@ -78,7 +79,8 @@ class PredictorCls(object):
         fcn_groups = []
         probabilities = []
         for ids in predict_cls:
-            cls_name = self.ids[ids]
+            position = self.id_list.index(ids)
+            cls_name = self.name_list[position]
             prob = result[ids]
             fcn_groups.append(cls_name)
             probabilities.append(prob)