validating the trigger event

model.py

@@ -180,14 +180,17 @@ Category:"""
         return predictions
 
     def fit(self, event, data, **kwargs):
+        """Handle annotation events from Label Studio
+        
+        Args:
+            event (str): Event type ('ANNOTATION_CREATED', 'ANNOTATION_UPDATED', 'START_TRAINING')
+            data (dict): Event payload with annotation details
         """
-        This method is called each time an annotation is created or updated
-        You can run your logic here to update the model and persist it to the cache
-        It is not recommended to perform long-running operations here, as it will block the main thread
-        Instead, consider running a separate process or a thread (like RQ worker) to perform the training
-        :param event: event type can be ('ANNOTATION_CREATED', 'ANNOTATION_UPDATED', 'START_TRAINING')
-        :param data: the payload received from the event (check [Webhook event reference](https://labelstud.io/guide/webhook_reference.html))
-        """
-        logger.info("Fit method called but not implemented in this version")
-        pass
+        # Log and validate event type
+        logger.info(f"Received event: {event}")
+        
+        valid_events = {'ANNOTATION_CREATED', 'ANNOTATION_UPDATED', 'START_TRAINING'}
+        if event not in valid_events:
+            logger.warning(f"Skip training: event {event} is not supported")
+            return