from typing import Dict, List, Any
import en_gendered_job_advertisements


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

        self.model = en_gendered_job_advertisements.load()

        # Preload all the elements you are going to need at inference.
        # pseudo:
        # self.model= load_model(path)

    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
        inputs = data.pop("inputs",data)

        prediction = self.model(inputs)

        return prediction