b2u commited on
Commit
b379c6b
·
1 Parent(s): dd6ec47

debugging prompt issue

Browse files
Files changed (1) hide show
  1. model.py +11 -16
model.py CHANGED
@@ -114,30 +114,25 @@ class BertClassifier(LabelStudioMLBase):
114
  formatted_text = self.instruction_template.format(text=cleaned_text)
115
  logger.debug(f"Original text: {original_text}")
116
  logger.debug(f"Cleaned text: {cleaned_text}")
 
117
  texts.append(formatted_text)
118
  except Exception as e:
119
- logger.error(f"Error formatting text: {str(e)}")
 
120
  raise
121
 
122
  model_predictions = self._model(texts)
123
- logger.debug(f"Raw model predictions: {model_predictions}")
124
-
125
  predictions = []
126
  for prediction in model_predictions:
127
- logger.debug(f"Processing prediction: {prediction}")
128
- try:
129
- region = li.get_tag(from_name).label(prediction['label'])
130
- pv = PredictionValue(
131
- score=prediction['score'],
132
- result=[region],
133
- model_version=self.get('model_version')
134
- )
135
- predictions.append(pv)
136
- except Exception as e:
137
- logger.error(f"Error processing prediction {prediction}: {str(e)}")
138
- raise
139
 
140
- logger.debug(f"Final predictions: {predictions}")
141
  return ModelResponse(predictions=predictions)
142
 
143
  except Exception as e:
 
114
  formatted_text = self.instruction_template.format(text=cleaned_text)
115
  logger.debug(f"Original text: {original_text}")
116
  logger.debug(f"Cleaned text: {cleaned_text}")
117
+ logger.debug(f"Formatted text: {formatted_text}")
118
  texts.append(formatted_text)
119
  except Exception as e:
120
+ logger.error(f"Error formatting text. Original text: {original_text}")
121
+ logger.error(f"Error details: {str(e)}")
122
  raise
123
 
124
  model_predictions = self._model(texts)
 
 
125
  predictions = []
126
  for prediction in model_predictions:
127
+ logger.debug(f"Prediction: {prediction}")
128
+ region = li.get_tag(from_name).label(prediction['label'])
129
+ pv = PredictionValue(
130
+ score=prediction['score'],
131
+ result=[region],
132
+ model_version=self.get('model_version')
133
+ )
134
+ predictions.append(pv)
 
 
 
 
135
 
 
136
  return ModelResponse(predictions=predictions)
137
 
138
  except Exception as e: