Update handler.py

handler.py

@@ -45,8 +45,7 @@ class EndpointHandler():
 
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         image = data['inputs'].pop("image", None)
+        type = data['inputs'].pop("type", '<MORE_DETAILED_CAPTION>')
+        text = data['inputs'].pop("text", None)
         image = Image.open(BytesIO(base64.b64decode(image)))
-
-        caption = self.run_example(image, '<MORE_DETAILED_CAPTION>')
-        ocr = self.run_example(image, '<OCR>')
-        return {**caption, **ocr}
+        return self.run_example(image, type, text_input=text)