Update app.py

app.py

@@ -9,12 +9,8 @@ from huggingface_hub import HfFileSystem
 from flax.serialization import msgpack_restore, from_state_dict
 import os
 
-
 hf_key =  text_input = st.text_input("Access token")
 
-
-
-
 class CNN(nn.Module):
   """A simple CNN model."""
 
@@ -35,28 +31,10 @@ class CNN(nn.Module):
 cnn = CNN()
 params = cnn.init(jax.random.PRNGKey(0), jnp.ones([1, 28, 28, 1]))['params']
 
-
 fs = HfFileSystem(token=hf_key)
 with fs.open("PrakhAI/HelloWorld/checkpoint.msgpack", "rb") as f:
     params = from_state_dict(params, msgpack_restore(f.read())["params"])
 
-
-
-# print(type(state))
-# print(state)
-# print([(k, type(v)) for (k, v) in state.items()])
-# print(state['params'])
-# print(state['opt_state'])
-
-# logits = state.apply_fn({'params': state.params}, batch['image'])
-# print(logits)
-
-# x = st.slider('Select a value')
-# st.write(x, 'squared is', x * x)
-
-# print(dir(cnn))
-
-
 uploaded_files = st.file_uploader("Input Images of handwritten digit (example in files)", type=['jpg','png','tif'], accept_multiple_files=True)
 
 if len(uploaded_files) == 0: