Spaces:
Build error
Build error
Update main.py
Browse files
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
|