inference.py

import joblib, pandas as pd
model = joblib.load('best_model.joblib')
def predict(df): return model.predict(df)