Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from  PIL import Image
+from io import BytesIO
 import tensorflow as tf
 import matplotlib.pyplot as plt
 import numpy as np
@@ -9,7 +10,8 @@ model = tf.keras.models.load_model('modelo.h5')
 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()
-    x = np.array(Image.open(BytesIO(bytes_data)))
+    x = np.array(Image.open(BytesIO(bytes_data).convert('L')))
+    print(x.shape)
     x = x / 255.00
     plt.figure()
     plt.imshow(imagen, cmap=plt.cm.binary)