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