debugging

model.py

@@ -37,8 +37,28 @@ class BertClassifier(LabelStudioMLBase):
         
         # Parse label config to get categories
         from label_studio_ml.model import parse_config
-        self.categories = parse_config(label_config)['choices']
-        logger.info(f"Parsed categories: {self.categories}")
+        parsed_config = parse_config(label_config)
+        logger.info(f"Parsed config: {parsed_config}")
+        
+        # Extract categories from the parsed config
+        if not parsed_config:
+            raise ValueError("Label config parsing returned empty result")
+        
+        # Find the first Choices tag in the config
+        choices_tag = None
+        for tag_name, tag_info in parsed_config.items():
+            if tag_info.get('type') == 'Choices':
+                choices_tag = tag_info
+                break
+        
+        if not choices_tag:
+            raise ValueError("No Choices tag found in label config")
+        
+        self.categories = choices_tag.get('values', [])
+        if not self.categories:
+            raise ValueError("No categories found in label config")
+        
+        logger.info(f"Found categories: {self.categories}")
         
         # Load training configuration from environment variables with defaults
         self.learning_rate = float(os.getenv('LEARNING_RATE', '2e-5'))