Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
+from flax.serialization import msgpack_restore
 
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)
 
-
+msgpack_restore("PrakhAI/HelloWorld/checkpoint.msgpack")