handling double webhooks

model.py

@@ -194,8 +194,16 @@ class T5Model(LabelStudioMLBase):
         
         try:
             # Extract text and label
-            text = data['task']['data']['text']
-            label = data['annotation']['result'][0]['value']['choices'][0]
+            # LS sends two webhooks when training is initiated:
+            # 1. contains all project data
+            # 2. contains only the task data
+            # We need to check which one is present and use the appropriate data
+            if 'task' in data:
+                text = data['task']['data']['text']
+                label = data['annotation']['result'][0]['value']['choices'][0]
+            else:
+                logger.info("Skipping initial project setup webhook")
+                return
             
             # Configure LoRA
             lora_config = LoraConfig(