from typing import Dict, List, Any from transformers import pipeline import holidays import sys import os class EndpointHandler: def __init__(self, path=""): self.pipeline = pipeline("text-classification", model=path) self.holidays = holidays.US() def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: """ data args: inputs (:obj: `str`) date (:obj: `str`) Return: A :obj:`list` | `dict`: will be serialized and returned """ return [{'data': '42'}, {'data2': '43'}]