jjgilces commited on
Commit
51d2d8a
·
1 Parent(s): a4ca81e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -66,6 +66,7 @@ def predict(img):
66
  return "Sin imagen", "Por favor, sube una imagen"
67
  # Cambia el tamaño de la imagen al tamaño de entrada esperado y prepárala para la predicción
68
  img = tf.image.resize(img, (224, 224))
 
69
  img = tf.expand_dims(img, axis=0)
70
 
71
  # Aquí, suponemos que tienes una lista de nombres de clases en el orden apropiado
 
66
  return "Sin imagen", "Por favor, sube una imagen"
67
  # Cambia el tamaño de la imagen al tamaño de entrada esperado y prepárala para la predicción
68
  img = tf.image.resize(img, (224, 224))
69
+ img = img / 255.0 # Escalar los valores de píxeles al rango [0, 1]
70
  img = tf.expand_dims(img, axis=0)
71
 
72
  # Aquí, suponemos que tienes una lista de nombres de clases en el orden apropiado