a bit debug

ocr_qa_assessment.py

@@ -14,8 +14,8 @@ class QAAssessmentPipeline(Pipeline):
         return text
 
     def _forward(self, text, **kwargs):
-        predictions, probabilities = self.model(text)
-        return predictions, probabilities
+        predictions = self.model(text)
+        return predictions
 
     def postprocess(self, outputs, **kwargs):
         predictions = outputs