fixing category retrival

model.py

@@ -54,7 +54,8 @@ class BertClassifier(LabelStudioMLBase):
         if not choices_tag:
             raise ValueError("No Choices tag found in label config")
         
-        self.categories = choices_tag.get('values', [])
+        # Extract labels from the choices tag
+        self.categories = choices_tag.get('labels', [])
         if not self.categories:
             raise ValueError("No categories found in label config")