inference.py

import joblib
import numpy as np
from PIL import Image

# Cargar el modelo
model = joblib.load("decision_tree_model.joblib")

# Función para cargar y aplanar una imagen
def load_and_flatten_image(image_path):
    img = Image.open(image_path)
    img = img.resize((28, 28))
    img_array = np.array(img)
    img_flat = img_array.flatten()
    return img_flat

def predict(image_path):
    img_flat = load_and_flatten_image(image_path)
    prediction = model.predict([img_flat])
    return prediction[0]