Update app.py

app.py

@@ -7,6 +7,10 @@ from flax.training import train_state  # Useful dataclass to keep train state
 from flax import linen as nn  # Linen API
 from huggingface_hub import HfFileSystem
 from flax.serialization import msgpack_restore, from_state_dict
+import os
+
+
+hf_key = os.environ['HF_KEY']
 
 
 
@@ -31,7 +35,7 @@ cnn = CNN()
 params = cnn.init(jax.random.PRNGKey(0), jnp.ones([1, 28, 28, 1]))['params']
 
 
-fs = HfFileSystem()
+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"])