Update app.py

app.py

@@ -57,17 +57,18 @@ with fs.open("PrakhAI/HelloWorld/checkpoint.msgpack", "rb") as f:
 # print(dir(cnn))
 
 
-uploaded_file = st.file_uploader("Input Image of handwritten digit (example in files)", type=['jpg','png','tif'], accept_multiple_files=False)
+uploaded_files = st.file_uploader("Input Images of handwritten digit (example in files)", type=['jpg','png','tif'], accept_multiple_files=True)
 
-if uploaded_file is None:
+if len(uploaded_files) == 0:
     st.write("Please upload an image!")
 else:
-    img = Image.open(uploaded_file)
-    rescaled = img.convert("HSV").split()[2].resize((28, 28))
-    st.image(rescaled)
-    brightness = jnp.array(rescaled)
-    input = brightness.reshape(1, 28, 28, 1) / 255.
-    st.write(cnn.apply({"params": params}, input).argmax(axis=1)[0])
+    for uploaded_file in uploaded_files:
+        img = Image.open(uploaded_file)
+        rescaled = img.convert("HSV").split()[2].resize((28, 28))
+        st.image(rescaled)
+        brightness = jnp.array(rescaled)
+        input = brightness.reshape(1, 28, 28, 1) / 255.
+        st.write(cnn.apply({"params": params}, input).argmax(axis=1)[0])
 
 def gridify(kernel, grid, kernel_size, scaling=5, padding=1):
     grid = np.pad(np.array(np.pad(np.repeat(np.repeat(kernel, repeats=scaling, axis=0), repeats=scaling, axis=1), ((padding,),(padding,),(0,),(0,)), 'constant', constant_values=(-1,)).reshape((kernel_size[0]*scaling+2*padding, kernel_size[1]*scaling+2*padding, grid[0], grid[1])).transpose(2,0,3,1).reshape(grid[0]*(kernel_size[0]*scaling+2*padding), grid[1]*(kernel_size[1]*scaling+2*padding))+1)*127., (padding,), 'constant', constant_values=(0,))