| 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] | |