Update app.py

app.py

@@ -8,6 +8,8 @@ with fs.open("PrakhAI/HelloWorld/checkpoint.msgpack", "rb") as f:
     state = msgpack_restore(f.read())
 
 print(type(state))
+print(state)
+print(state['params'])
 
 # logits = state.apply_fn({'params': state.params}, batch['image'])
 # print(logits)