File size: 497 Bytes
fed5c73 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | from abc import abstractmethod
from typing import Any, Tuple, List, Dict, Union
from src.misc.schemas import ProductNamedEntityExtractionSchema
class BaseModel:
def __init__(self) -> None:
pass
def __call__(
self, input_query: str, *args: Any, **kwds: Any
) -> ProductNamedEntityExtractionSchema:
return self.predict(input_query=input_query)
@abstractmethod
def predict(self, input_query, *args: Any, **kwds: Any):
raise NotImplementedError
|