unnecessary logs removed

model.py

@@ -113,12 +113,12 @@ class BertClassifier(LabelStudioMLBase):
         if not self._model or not self.tokenizer:
             logger.error("Model or tokenizer not initialized")
             return []
-        
+            
         # Check if categories match the Label Studio config
         if not hasattr(self, 'categories') or not self.categories:
             logger.error("No categories configured")
             return []
-        
+            
         logger.info("=== INPUT VALIDATION ===")
         logger.info(f"Number of tasks: {len(tasks)}")
         logger.info(f"Model loaded: {self._model is not None}")
@@ -158,8 +158,6 @@ class BertClassifier(LabelStudioMLBase):
                     logger.error(f"Task {task_index}: Empty tokenized input")
                     continue
                 
-                logger.info(f"Tokenized length: {inputs['input_ids'].size(1)}")
-                
                 # Get model prediction
                 self._model.eval()
                 with torch.no_grad():
@@ -181,13 +179,6 @@ class BertClassifier(LabelStudioMLBase):
                         
                     confidence_score = top_probs[0][0].item()
                 
-                # Validate prediction
-                if confidence_score < 0.1:
-                    logger.warning(f"Low confidence prediction: {confidence_score:.4f}")
-                
-                logger.info(f"Predicted labels: {choices}")
-                logger.info(f"Top confidence score: {confidence_score:.4f}")
-                
                 # Format prediction according to Label Studio requirements
                 prediction = {
                     'result': [{
@@ -208,7 +199,6 @@ class BertClassifier(LabelStudioMLBase):
                     logger.error(f"Task {task_index}: Invalid prediction format")
                     continue
                     
-                logger.info(f"Formatted prediction: {json.dumps(prediction, indent=2)}")
                 predictions.append(prediction)
                 
             except Exception as e: