File size: 630 Bytes
3806a86 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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ı"}] |