Update app.py

app.py

@@ -7,6 +7,17 @@ import numpy as np
 
 model = tf.keras.models.load_model('modelo.h5')
 
+labels = ['Camiseta',
+ 'Pantalon',
+ 'Pull-over',
+ 'Vestido',
+ 'Abrigo',
+ 'Sandalia',
+ 'Camisa',
+ 'Zapato',
+ 'Bolso',
+ 'Bota']
+
 file = st.file_uploader("Por favor suba una imagen de ropa (jpg, png, jpeg)", type=['jpg','png','jpeg'])
 if file is not None:
     bytes_data = file.read()
@@ -18,16 +29,16 @@ if file is not None:
     x = np.squeeze(x)
     x  = np.expand_dims(x, axis=0)
     prediction = model.predict(x)
-    print(prediction)
-    
+    index = numpy.argmax(prediction[0])
+    label = labels[index]
+    text = f"<p style='text-align: center;'>{label}</p>"
     col1, col2, col3 = st.columns(3)
-
     with col1:
         st.write("")
     
     with col2:
         image = Image.open(file)
-        st.markdown('<p style="text-align: center;">Image</p>',unsafe_allow_html=True)
+        st.markdown(text,unsafe_allow_html=True)
         st.image(image,width=300)  
     
     with col3: