.

model.py

@@ -140,7 +140,17 @@ class T5Model(LabelStudioMLBase):
         logger.info(f"Additional kwargs: {kwargs}")
         
         predictions = []
-        valid_choices = self.get_valid_choices(kwargs.get('label_config', ''))
+        # Get valid choices during initialization instead of from kwargs
+        valid_choices = []
+        try:
+            import xml.etree.ElementTree as ET
+            root = ET.fromstring(self.label_config)
+            choices = root.findall('.//Choice')
+            valid_choices = [choice.get('value') for choice in choices]
+            logger.info(f"Valid choices: {valid_choices}")
+        except Exception as e:
+            logger.error(f"Error parsing choices: {str(e)}")
+            valid_choices = ["no_category"]  # fallback
         
         try:
             for task in tasks: