Update app.py

app.py

@@ -11,8 +11,13 @@ file = st.file_uploader("Por favor suba una imagen de ropa (jpg, png, jpeg)", ty
 if file is not None:
     bytes_data = file.read()
     x = np.array(Image.open(BytesIO(bytes_data)).convert('L'))
-    print(x.shape)
     x = x / 255.00
+    x  = np.expand_dims(x, axis=-1)
+    print(x.shape)
+    x  = tf.image.resize(x [80, 80])
+    print(x.shape)
+    x = np.repeat(x[:, :, np.newaxis], 3, axis=2)
+    print(x.shape)
     plt.figure()
     plt.imshow(x, cmap=plt.cm.binary)
     plt.colorbar()