webhook fix

model.py

@@ -69,10 +69,28 @@ class BertClassifier(LabelStudioMLBase):
     finetuned_model_name = os.getenv('FINETUNED_MODEL_NAME', 'finetuned-model')
     _model = None
 
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-        # Simplest default - just the placeholder
+    def __init__(self, project_id=None, label_config=None, **kwargs):
+        # Initialize parent class properly
+        super(BertClassifier, self).__init__(project_id=project_id, label_config=label_config)
+        
+        # Your existing initialization code
+        self.label_encoder = LabelEncoder()
+        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.instruction_template = os.getenv('MODEL_INSTRUCTIONS', '{text}')
+        
+        # Define your categories
+        self.categories = [
+            'affiliate_classification', 'brand', 'business_and_career',
+            'content_quality', 'date', 'demographic', 'event',
+            'faith_and_religion', 'gaming', 'health',
+            'internal_categorization', 'location', 'number',
+            'performance', 'post_type', 'pricing_tier',
+            'product', 'profession', 'pii', 'social_network',
+            'style_and_fashion', 'no_category'
+        ]
+        
+        # Fit label encoder with your categories
+        self.label_encoder.fit(self.categories)
 
     def get_labels(self):
         li = self.label_interface