Update handler.py

handler.py

@@ -12,16 +12,19 @@ class EndpointHandler():
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
        data args:
-            images (:obj:`string`)
-            candiates (:obj:`list`)
+            parameters: {
+                candidate_labels: List[str]
+            }
+            inputs: str
       Return:
             A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82}
         """
-        inputs = data.pop("inputs", data)
+        parameters = data.get("parameters", {})
+        inputs = data.get("inputs", "")
 
         # decode base64 image to PIL
-        image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
+        image = Image.open(BytesIO(base64.b64decode(inputs)))
 
         # run prediction one image wit provided candiates
-        prediction = self.pipeline(images=[image], candidate_labels=inputs["candiates"])
+        prediction = self.pipeline(images=[image], candidate_labels=parameters.get("candidate_labels", []))
         return prediction[0]