INNER CODE UNIT · Python

__call__

maxpumperla/learning_ray · notebooks/app.py:93

    async def __call__(self, input_text: str) -> bool:
        return await self.is_positive_batched(input_text)


@serve.deployment(num_replicas=2)
class Summarizer:
    def __init__(self, max_length: Optional[int] = None):
        self._summarizer = pipeline("summarization")
        self._max_length = max_length

    def __call__(self, input_text: str) -> str:
        result = self._summarizer(
            input_text, max_length=self._max_length, truncation=True)
        return result[0]["summary_text"]


@serve.deployment
class EntityRecognition:

View source record →