Update app.py

app.py

@@ -13,16 +13,15 @@ if file is not None:
     x = np.array(Image.open(BytesIO(bytes_data)).convert('L'))
     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()
     plt.grid(False)
     plt.show()
+    x = np.repeat(x[:, :, np.newaxis], 3, axis=2)
+    x = np.squeeze(x)
+
     
     col1, col2, col3 = st.columns(3)