Update app.py

app.py

@@ -9,8 +9,6 @@ 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."""
 
@@ -31,7 +29,7 @@ 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)
+fs = HfFileSystem()
 with fs.open("PrakhAI/HelloWorld/checkpoint.msgpack", "rb") as f:
     params = from_state_dict(params, msgpack_restore(f.read())["params"])