fashxp
/

zero-shot-image-classification

Zero-Shot Image Classification

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

fashxp commited on Jan 25

Commit

8852310

•

1 Parent(s): 6617b85

updates

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -12,12 +12,12 @@ class EndpointHandler():
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
        data args:
-            image (:obj:`string`)
             parameters (:obj:)
       Return:
             A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82}
         """
-        image_data = data.pop("image", data)
         # decode base64 image to PIL
         image = Image.open(BytesIO(base64.b64decode(image_data)))

     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
        data args:
+            inputs (:obj:`string`)
             parameters (:obj:)
       Return:
             A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82}
         """
+        image_data = data.pop("inputs", data)
         # decode base64 image to PIL
         image = Image.open(BytesIO(base64.b64decode(image_data)))