a bit debug

ocr_qa_assessment.py

@@ -18,12 +18,10 @@ class QAAssessmentPipeline(Pipeline):
         return predictions, probabilities
 
     def postprocess(self, outputs, **kwargs):
-        predictions, probabilities = outputs
+        predictions = outputs
+        print(f"Predictions: {predictions}")
         label = predictions[0][0].replace("__label__", "")  # Remove __label__ prefix
-        confidence = float(
-            probabilities[0][0]
-        )  # Convert to float for JSON serialization
 
         # Format as JSON-compatible dictionary
-        model_output = {"label": label, "confidence": round(confidence * 100, 2)}
+        model_output = {"label": label, "score": round(score, 4)}
         return model_output