Update handler.py

handler.py

@@ -33,4 +33,4 @@ class EndpointHandler:
         else:
             prediction = self.pipline(inputs)
         # postprocess the prediction
-        return prediction
+        return [{"label": p["label"]} for p in prediction]