19b3da3
|
|
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)
|
