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]