Spaces:
Build error
Build error
File size: 584 Bytes
d660b02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from abc import ABC, abstractmethod
class DeploymentStrategy(ABC):
@abstractmethod
def deploy(self, model, endpoint_name: str, endpoint_config_name: str) -> None:
pass
class Inference(ABC):
"""An abstract class for performing inference."""
def __init__(self):
self.model = None
@abstractmethod
def set_payload(self, inputs, parameters=None):
pass
@abstractmethod
def set_payload(self, query, context, parameters=None):
pass
@abstractmethod
def inference(self):
pass
|