jeff-RQ
/

new-test-model

image-captioning

visual-question-answering

Model card Files Files and versions

jeff-RQ commited on Jul 4, 2023

Commit

40bec71

·

1 Parent(s): 5cce52e

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -15,7 +15,7 @@ class EndpointHandler:
     def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
         # process input
-        # image = data.pop("image", data)
         text = data.pop("text", data)
         image_string = base64.b64decode(data["image"])

     def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
         # process input
+        data = data.pop("inputs", data)
         text = data.pop("text", data)
         image_string = base64.b64decode(data["image"])