handler.py

from typing import Dict, List, Any
from keras.models import load_model

class EndpointHandler():
    def __init__(self, path="sepres-20-epoch-sigmoid.h5"):
        self.model = load_model(path)

    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
        return {'output': 'test'}