Update app.py

app.py

@@ -45,6 +45,7 @@ else:
     img = Image.open(uploaded_file)
     st.image(img)
     input = jnp.array(tf.cast(tf.image.resize(tf.convert_to_tensor(img), [50, 50]), tf.float32) / 255.)
+    st.write("Input shape: " + input.shape)
     st.write("Model Prediction: " + cnn.apply({"params": params}, input))
     st.write("Model Prediction type: " + type(cnn.apply({"params": params}, input)))
     st.write("Model Prediction type dir: " + dir(cnn.apply({"params": params}, input)))