from typing import Any, Dict, List

from inference import model_fn, predict_fn


class EndpointHandler:
    def __init__(self, path=""):
        return model_fn(path)

    def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
        return predict_fn(data, None)