Update app.py

app.py

@@ -38,11 +38,6 @@ uploaded_files = st.file_uploader("Input Images of handwritten digit (example in
 if len(uploaded_files) == 0:
     st.write("Please upload an image!")
 else:
-    for uploaded_file in uploaded_files:
-        rescaled = Image.open(uploaded_file).convert("HSV").split()[2].resize((28, 28))
-        st.image(rescaled)
-        input = jnp.array(rescaled).reshape(1, 28, 28, 1) / 255.
-        st.write("Model Prediction: " + cnn.apply({"params": params}, input).argmax(axis=1)[0])
   input = jnp.array([jnp.array(Image.open(uploaded_file).convert("HSV").split()[2].resize((28, 28))).reshape(1, 28, 28, 1) / 255. for uploaded_file in uploaded_files])
   prediction = cnn.apply({"params": params}, input)
   for (index, image) in enumerate(uploaded_files):