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