inference.py

def model_fn(model_dir):
    import joblib
    import os
    return joblib.load(os.path.join(model_dir, "sklearn_model.joblib"))

def predict_fn(input_data, model):
    return model.predict(input_data)