inference.py

import joblib
import numpy as np

# Charger le modèle
model = joblib.load("model.joblib")

# Fonction de prédiction
def predict(input_data):
    """
    input_data : array-like ou liste de caractéristiques
    Retourne les prédictions du modèle
    """
    input_array = np.array(input_data).reshape(1, -1)
    return model.predict(input_array).tolist()