Upload inference.py with huggingface_hub

inference.py

@@ -0,0 +1,19 @@
+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]