Update app.py

app.py

@@ -9,9 +9,9 @@ with fs.open("PrakhAI/HelloWorld/checkpoint.msgpack", "rb") as f:
 
 print(type(state))
 print(state)
-print(state.keys)
-print(state['params'])
-print(state['opt_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)