25 / handler.py
Huseyin's picture
Create handler.py
3806a86 verified
from typing import Dict, List, Any
import torch
class EndpointHandler:
def __init__(self, path=""):
# Modelinizi yükleyin
# path parametresi model dosyalarının bulunduğu dizini gösterir
pass
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
"""
data: inputs anahtarı ile gelen veri
return: Model tahmin sonucu
"""
inputs = data.pop("inputs", data)
# Burada model inference işlemini yapın
# Örnek:
# predictions = self.model(inputs)
return [{"predictions": "model çıktısı"}]