| from typing import Dict, List, Any | |
| from io import BytesIO | |
| import base64 | |
| import logging | |
| import uform | |
| class EndpointHandler(): | |
| def __init__(self, path=""): | |
| self.model, self.processor = uform.get_model('unum-cloud/uform-vl-multilingual-v2') | |
| def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: | |
| """ | |
| data args: | |
| image (:obj:`string`) | |
| candidates (:obj:`list`) | |
| Return: | |
| A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82} | |
| """ |