Handler

• Validate the input and output data for your model
• Add pre and post-processing logic around your model
• Log inference results to a third-party tool, such as Weights and Biases
• Retrieve and update model embeddings
• Call third-party APIs

import slai


class ModelHandler(slai.BaseModelHandler):
    inputs = {"text": slai.types.String()}

    outputs = {"generated_text": slai.types.String(), "sentiment": slai.types.String()}

    # When serving the model, the model is first loaded into memory
    # The loaded model is then available in the "call" method
    def load(self, model=None):
        loaded_model = model
        return loaded_model

    # This is the entry point for your API
    def call(self, model, **inputs):
        gpt2_model = model["gpt2_model"]
        sentiment_model = model["sentiment_analysis_model"]

        gpt2_prediction = gpt2_model(
            inputs["text"], max_length=10, num_return_sequences=1
        )[0]["generated_text"]
        sentiment_prediction = sentiment_model(gpt2_prediction)[0]["label"]

        return {"generated_text": gpt2_prediction, "sentiment": sentiment_prediction}

• inputs: Defines the input data that will be passed to your API. This must be a dictionary containing the names and data types of your model inputs. In the example above, we are specifying that the inputs to this API are three float variables with the names: age, weight, and height
• outputs: Defines the output data that will be returned from your API. This must be a dictionary containing the names and data types of your model outputs. In the example above, we are specifying that the output from this model is one float with the name: prediction
• load: This function lets you perform any kind of operations you want to only happen once when your API is first deployed. You can also use this method to load a pre-trained model if you’re not using Slai for training. The model parameter contains the model artifact that was produced during training. If you did not train, this variable will be empty.
• call: The entry point to your API. In this function, you have access to any artifacts that were produced during training, as well as the inputs defined above. The return value of this function must be a dictionary matching the types specified in the outputs dictionary.

​

Input and Output Types