RougeVertBleu commited on
Commit
4e30ecb
·
verified ·
1 Parent(s): bd23ffd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import tensorflow as tf
3
  import numpy as np
4
- from PIL import Image
5
 
6
  # Charger les modèles (remplacez les chemins par vos fichiers de modèles)
7
  model1 = tf.keras.models.load_model('model_v2_Q1.keras')
@@ -17,7 +17,7 @@ def format_image_type1(image):
17
  return image_single_fix(image)
18
 
19
  def format_image_type2(image):
20
- image = load_img(image, color_mode='grayscale') # Charge en noir et blanc mais avec une meilleure résolution
21
  image = img.resize((128, 128), Image.LANCZOS) # Redimensionner selon la taille attendue
22
  image = np.array(img) / 255.0 # Normalisation
23
  return image_single_fix(image)
 
1
  import gradio as gr
2
  import tensorflow as tf
3
  import numpy as np
4
+ from PIL import Image, ImageOps
5
 
6
  # Charger les modèles (remplacez les chemins par vos fichiers de modèles)
7
  model1 = tf.keras.models.load_model('model_v2_Q1.keras')
 
17
  return image_single_fix(image)
18
 
19
  def format_image_type2(image):
20
+ image = ImageOps.grayscale(image) # Charge en noir et blanc mais avec une meilleure résolution
21
  image = img.resize((128, 128), Image.LANCZOS) # Redimensionner selon la taille attendue
22
  image = np.array(img) / 255.0 # Normalisation
23
  return image_single_fix(image)