Spaces:
Sleeping
Sleeping
debugging prompt issue
Browse files
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: {
|
|
|
|
| 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"
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
| 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:
|