utils.py

import joblib
import pandas as pd

def load_model(path: str):
    """Load a saved pipeline model from disk."""
    return joblib.load(path)


def predict(model, input_df: pd.DataFrame):
    """

    Predict class labels and probabilities.

    Returns: (labels, probabilities)

    """
    labels = model.predict(input_df)
    probs = model.predict_proba(input_df)
    return labels, probs