Docfile commited on
Commit
998fa4a
·
1 Parent(s): 4740eb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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.ANTIALIAS)
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", capture_session=True)
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()