Update pipeline.py

pipeline.py

@@ -17,9 +17,11 @@ class PreTrainedPipeline():
       Return:
             A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82}
         """
+        inputs = data.pop("inputs", data)
+
         # decode base64 image to PIL
-        image = Image.open(BytesIO(base64.b64decode(data['image'])))
+        image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
 
         # run prediction one image wit provided candiates
-        prediction = self.pipeline(images=[image], candidate_labels=data["candiates"])
+        prediction = self.pipeline(images=[image], candidate_labels=inputs["candiates"])
         return prediction[0]