Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,13 +10,12 @@ loaded_model = joblib.load(filename)
|
|
| 10 |
|
| 11 |
# Fonction de traitement d'image
|
| 12 |
def preprocess_image(image):
|
|
|
|
| 13 |
pil_image = Image.fromarray(image)
|
| 14 |
gray_image = pil_image.convert("L")
|
| 15 |
-
# Convertir l'image en niveaux de gris
|
| 16 |
-
#gray_image = image.convert("L")
|
| 17 |
|
| 18 |
# Redimensionner l'image à 8x8 pixels
|
| 19 |
-
resized_image = gray_image.resize((8, 8), Image.
|
| 20 |
|
| 21 |
# Convertir l'image en tableau NumPy
|
| 22 |
img_array = np.array(resized_image)
|
|
@@ -37,5 +36,5 @@ def predict_digit(image):
|
|
| 37 |
prediction = loaded_model.predict([preprocessed_image])[0]
|
| 38 |
return str(prediction)
|
| 39 |
|
| 40 |
-
iface = gr.Interface(fn=predict_digit, inputs="image", outputs="text"
|
| 41 |
iface.launch()
|
|
|
|
| 10 |
|
| 11 |
# Fonction de traitement d'image
|
| 12 |
def preprocess_image(image):
|
| 13 |
+
# Convertir l'image en niveaux de gris
|
| 14 |
pil_image = Image.fromarray(image)
|
| 15 |
gray_image = pil_image.convert("L")
|
|
|
|
|
|
|
| 16 |
|
| 17 |
# Redimensionner l'image à 8x8 pixels
|
| 18 |
+
resized_image = gray_image.resize((8, 8), Image.LANCZOS)
|
| 19 |
|
| 20 |
# Convertir l'image en tableau NumPy
|
| 21 |
img_array = np.array(resized_image)
|
|
|
|
| 36 |
prediction = loaded_model.predict([preprocessed_image])[0]
|
| 37 |
return str(prediction)
|
| 38 |
|
| 39 |
+
iface = gr.Interface(fn=predict_digit, inputs="image", outputs="text")
|
| 40 |
iface.launch()
|