Update app.py

app.py

@@ -5,12 +5,11 @@ import tensorflow as tf
  
 # Etiquetas en español
 cifar10_labels = np.array([
-    'avión', 'automóvil', 'pájaro', 'gato', 'venado',
-    'perro', 'rana', 'caballo', 'barco', 'camión'
+    'avión', 'automóvil', 'pájaro', 'gato', 'venado'
 ])
  
 # Cargar el modelo al iniciar la app
-model = tf.keras.models.load_model('my_model.keras')
+model = tf.keras.models.load_model('my_model.h5')
  
 def preprocess_image(image):
     """Preprocesa la imagen para el modelo"""
@@ -23,7 +22,7 @@ def predict(image):
     """Realiza la predicción y devuelve los resultados"""
     processed_img = preprocess_image(image)
     preds = model.predict(processed_img)[0]
-    return {cifar10_labels[i]: float(preds[i]) for i in range(10)}
+    return {cifar10_labels[i]: float(preds[i]) for i in range(5)}
  
 # Configuración de la interfaz
 title = "CIFAR-10"