Update handler.py

handler.py

@@ -31,7 +31,7 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
         prediction, prediction_prob = self.pipeline(inputs)
         # postprocess the prediction
-        return {"prediction": prediction, "prediction_prob": prediction_prob}  
+        return [{"prediction": prediction, "prediction_prob": prediction_prob}]
 
 
 class TextClassificationPipeline: