handler.py

from inference import Chat

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

    def __call__(self, inputs):
        return {"output": self.chat.answer(inputs)}