Update model.py

model.py

@@ -43,6 +43,7 @@ class Model(LabelStudioMLBase):
     os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = get_credentials()
     image_processor = AutoImageProcessor.from_pretrained("diegokauer/conditional-detr-coe-int")
     model = AutoModelForObjectDetection.from_pretrained("diegokauer/conditional-detr-coe-int")
+    id2label = model.config.id2label
 
     def predict(self, tasks, **kwargs):
         """ This is where inference happens: model returns