inferece.py

import joblib
import numpy as np

model = joblib.load("classifier.joblib")

def predict(inputs):
    X = np.array([[v for v in sample.values()] for sample in inputs])
    predictions = model.predict(X)
    return [int(p) for p in predictions]