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()
|